[adjkant]

I think you've missed a step jumping straight to prison.

If you accept a lack of free will, the solution does not need to immediately jump there but rather to finding a way that is fair to all to prevent the bad behavior thing from affecting others negatively while not locking someone up in a cage. There are many lines in between and some penal systems have adapted, but the US is way far behind there.

The optimization is no longer about revenge / punishment but about altering the scenario of the world to make everyone work together better. Sometimes people need to be fully separated from society, but not often. I think we do have effective behavior altering treatments, but they just aren't drugs and take time. But it's not a straight line from "it takes time to help people and its unreliable" to "we're giving up on finding a way for society to work with the people who did not choose what they are".