The Victorian government has admitted it needs to overhaul the way prisoners with mental illnesses and cognitive impairments are treated, concerned they are over-represented in jail, suffer more in custody and are more likely to reoffend on release.