The only problem with this is that it works in reverse. We could put people who haven't commited a crime in jail on the grounds that they are likely to or it helps society when their in jail.

Before things go that far, shouldn't a society set up voluntary programs for treatment? Exactly how does one draw the line between punishment and treatment? Our society has blurred the two notions. (Plea bargaining involving attendance of a driving course.)