The society has taken damage when the crime was executed, and by punishment the loss is alleviated.
Well, that is
another imbalance argument.
The problem I have with that is the idea that society is somehow "enriched" by meting out punishment.
From what I see, we have a double loss situation, for the (punished) criminal AND society, both gets poorer.
As I see it, that double loss is incurred at the moment of the crime.
To
punish justly, i.e, treating the criminal in a way according to his reduced status, but that would be inadmissible to treat a non-criminal might, possibly, offer some consolation to the victims/society t large, might induce some regrets in the criminal, but none of this provides the justification of punishment.
For example, a lot of criminals DO regret their actions sincerely&immensely, and would never think ever of doing anything like that again.
That does not, however, on its own, restore their status of equal legal worth with the non-criminal (and hence, inadmissibility of punishment).
If the society is to be said to be "enriched", it lies in exercising the greater scope of freedom of action upon the criminal that his self-inflicted loss of rights has given society.