A method and system for generating a treatment plan are disclosed herein. A computing system receives a plurality of dose volume histograms for a plurality of patients and a plurality of volumetric dose distributions corresponding to the plurality of dose volume histograms. The computing system generates a volumetric dose prediction model using a neural network by learning, by the neural network, a relationship between a plurality of dose volume histograms for the plurality of patients and the corresponding plurality of volumetric dose distributions. The computing system receives a candidate dose volume histogram for a target patient. The computing system infers, via the volumetric dose prediction module, a volumetric dose prediction distribution matching the candidate dose volume histogram. The computing system generates a recommendation based on the inferred volumetric dose prediction distribution.