The Dark MAtter Particle Explorer (DAMPE) instrument is a space-born cosmic-ray detector, capable of measuring ion fluxes up to $\sim$500 TeV/n. This energy scale is made accessible through its calorimeter, which is the deepest currently operating in orbit. Saturation of the calorimeter readout channels start occurring above $\sim$100 TeV of incident energy, and can significantly affect the primary energy reconstruction. Different techniques, analytical and machine-learning based, have been developed to tackle this issue, focusing on the recovery of single-bar deposits, up to several hundreds of TeV. In this work, a new machine-learning technique is presented, which profit of a unique model to correct the total deposited energy in DAMPE calorimeter. The described method is able to generalise its corrections for different ions and extend the maximum detectable incident energy to the PeV scale.