The Dark MAtter Particle Explorer (DAMPE) instrument is a space-borne cosmic-ray detector, capable of measuring ion fluxes up to ∼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 starts occurring above ∼100 TeV of incident energy, and can significantly affect the primary energy reconstruction. Different techniques-analytical and machine-learning based-were 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 benefits from 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. This work is a continuation of the results presented in Stolpovskiyet al. (2022).