I think the best option would be to increase the 'heal rates' for units.

Also, when the computer AI routs, you immediately get the END BATTLE screen. If the player continues, he can easily decimate the routed armies.

Maybe we could increase the speed of routing troops so that they cant be caught so easily. Combining this with increased heal rates should give survivors a pretty decent increase in troops surviving