The optimal interaction of drugs with plasma membranes and membranes of subcellular organelles is a prerequisite for desirable pharmacology. Importantly, for drugs targeting the transmembrane lipid-facing sites of integral membrane proteins, the relative affinity of a drug to the bilayer lipids compared to the surrounding aqueous phase affects the partitioning, access, and binding of the drug to the target site. Molecular dynamics (MD) simulations, including enhanced sampling techniques such as steered MD, umbrella sampling (US), and metadynamics, offer valuable insights into the interactions of drugs with the membrane lipids and water in atomistic detail. However, these methods are computationally prohibitive for the high-throughput screening of drug candidates. This study shows that applying denoising diffusion probabilistic models (DDPMs), a generative AI method, to US simulation data reduces the computational cost significantly. Specifically, the models used only partial (one-third) data from the US simulations and reproduced the complete potential of mean force (PMF) profiles for three FDA-approved drugs (β2-adrenergic agonists) and ∼20 biologically relevant chemicals with known experimentally characterized bilayer locations. Intriguingly, the model can predict the solvation-free energies for partitioning and crossing the bilayer, preferred bilayer locations (low-energy well), and orientations of the ligands with high accuracy. The results indicate that DDPMs can be used to characterize the complete membrane partitioning profile of drug molecules using fewer umbrella sampling simulations at select positions along the bilayer normal (z-axis), irrespective of their amphiphilic−lipophilic−cephalophilic characteristics.