Not sure if possible, might require hard code.
I am thinking on permanently reducing strength of unit when individual figurines are dead.
On the oppsite side, here should be option (skill for all trained unit) to rebeuild.
This would make the unit inactive for 1 turn and restore a squad figurine.
this could however cripple the AI .... might be not a good mod idea still