A novel diffusion-based method called DIME is introduced for learning the joint distribution of multiple interdependent medical treatment outcomes. DIME addresses the limitations of existing machine learning approaches by capturing dependence structures and handling mixed outcome types, thereby enabling more reliable decision-making with uncertainty quantification. Experimental results demonstrate its effectiveness in learning the multi-outcome distribution of medical treatments.
machine-learning ✓
medical ✓
+ outcomes
diffusion ✓
causal-inference ✓