Few areas of computational chemistry are moving faster than diffusion models for drug design, which borrow the same class of artificial intelligence (AI) method that powers modern image generation and point it at molecules. The core idea is elegant: start from random noise and progressively refine it into a structured molecule, learning the reverse of a process that gradually destroys structure. What makes this genuinely useful for drug design is that these models can work directly in three dimensions, generating a molecule's actual 3D shape, and even generating it inside a target protein's binding pocket, rather than producing a flat chemical formula and inferring the shape afterward.
This article explains the technical principles for a scientifically literate reader and assesses what the models deliver in practice. It sits within DDN's coverage of generative AI in molecular design, complements the discussion of AlphaFold and protein structure prediction, and connects to the broader guide to AI across drug discovery.
Key takeaways
|
What is a diffusion model?
A diffusion model is a type of generative AI that learns to create data by reversing a process of gradual corruption. The idea is easiest to grasp through its now-familiar use in images: during training, the model takes a clear image and adds random noise to it step by step until nothing but noise remains, then learns to reverse that process, predicting how to remove noise at each step. Once trained, it can start from pure noise and denoise its way to a brand-new, coherent image. The same mathematics applies to molecules, with 3D atomic structure taking the place of pixels.
The core mechanics, in plain terms:
- The forward process. Structure is progressively destroyed by adding noise in small steps, turning a well-formed molecule into a random cloud of atoms. This step is used only in training and is straightforward to define.
- The reverse process. The model learns to undo the corruption one step at a time, and this learned denoising is the generative engine: applied to fresh noise, it produces a new structured molecule.
- Equivariance. For molecules, the model must respect the physics of 3D space, so that rotating or translating a molecule does not change its chemistry. Diffusion models for molecules use equivariant neural networks that build this symmetry in, which is central to why they work well in 3D.
- Conditioning. Generation can be guided by a condition, most importantly the structure of a target protein's binding pocket, so the model generates molecules tailored to a specific target rather than sampling chemical space at random.
The reason diffusion models displaced earlier approaches so quickly is largely practical: they train more stably and produce higher-quality, more diverse output than the generative methods that preceded them, a difference that turns out to matter a great deal for molecular generation.
How diffusion is applied to molecular generation
Translating the diffusion recipe from images to molecules requires deciding what exactly gets noised and denoised, and the answer shapes what a model can do. The most powerful approaches for drug design operate on the full 3D structure: the atoms, their types and their positions in space, and often the bonds between them.
What a molecular diffusion model generates:
- Atom types and coordinates. The model jointly denoises which atoms are present and where they sit in 3D space, building a molecule as a set of positioned atoms rather than a text string, which keeps geometry central throughout generation.
- Bonds and connectivity. Because atom positions alone do not fully specify a molecule, better models also generate the bonds explicitly, modeling the interdependence of atoms and bonds so the output is a chemically coherent structure rather than a plausible-looking point cloud.
- Pocket-conditioned generation. For structure-based design, the target protein's binding pocket is supplied as a fixed condition, and the model generates a ligand within it, so the molecule is shaped to the site from the first denoising step.
- Fragment and partial generation. By fixing part of a structure and denoising the rest, diffusion models support practical medicinal chemistry tasks such as fragment growing, scaffold hopping and linker design, not just whole-molecule generation from scratch.
That last capability matters more than it might seem. The ability to fix a known-good part of a molecule and regenerate the rest maps directly onto how medicinal chemists actually work, refining a lead rather than starting over, which is part of why 3D diffusion has drawn serious interest from drug design programs rather than remaining a purely academic technique.
3D vs. 2D generation
A central distinction in generative molecular design is whether a model works in two dimensions, generating a molecule as a graph or text string, or in three, generating its actual spatial structure. Diffusion models are not the only way to do 3D generation, but they are the approach that made high-quality 3D generation practical, and the difference has real consequences for drug design.
Dimension | What the model generates | Consequence for drug design |
2D (graph/string) | Atoms and bonds as a topology, no coordinates | Fast and valid, but shape and fit must be inferred separately |
3D (coordinates) | Atoms positioned in space, often with bonds | Designs for physical fit directly, at higher computational cost |
3D, pocket-conditioned | A ligand generated inside the target's binding site | Accounts for the target's shape from the start of generation |
The case for 3D is that binding is fundamentally a three-dimensional phenomenon: whether a molecule fits and binds its target depends on shape and spatial complementarity that a 2D representation does not capture. Generating directly in 3D, especially inside the pocket, lets the model reason about fit while it designs. The cost is computational expense and, as the benchmarking discussion below shows, a tendency to produce geometrically plausible molecules that are not always chemically sensible.
Binding is a three-dimensional phenomenon. A molecule that looks right on paper may not fit its target in space, which is the case for generating molecules in 3D, and also the source of diffusion models' hardest problem: 3D-plausible is not the same as chemically real.
Key models: DiffDock, DiffSBDD, TargetDiff
Three models are worth understanding because they represent the main ways diffusion is applied in structure-based drug design, and because they anchor much of the benchmarking literature. Each takes a distinct angle on the problem.
The landmark diffusion models for drug design:
- DiffDock. Rather than generating a molecule, DiffDock reframes molecular docking as a generative task, using diffusion to produce ligand binding poses along with confidence estimates. It demonstrated substantially improved performance over existing state-of-the-art docking methods on a common benchmark while running relatively quickly, a notable result for the docking problem specifically.
- DiffSBDD. A structure-based design model, DiffSBDD uses an equivariant diffusion approach to design molecules directly in the 3D environment of a protein's binding site, denoising sets of points into ligands conditioned on the pocket. It also supports inpainting-style partial generation for fragment and scaffold tasks.
- TargetDiff. A 3D equivariant model, TargetDiff jointly generates atom coordinates and types conditioned on protein pocket atoms, and additionally supports binding-affinity prediction from its learned representations, performing strongly on a standard structure-based benchmark in both molecular geometry and binding metrics.
The distinction among them is instructive. DiffDock addresses where a known molecule binds; DiffSBDD and TargetDiff address what molecule to make for a given pocket. Both questions matter in a drug program, and diffusion has made credible progress on each, which is why these models are reference points rather than curiosities.
Benchmarking
Benchmarking is where enthusiasm meets evidence, and for diffusion models in drug design the picture is genuinely mixed. The models perform well on the metrics they were designed around, geometry and predicted binding, while independent evaluation has repeatedly exposed weaknesses in the properties that actually determine whether a generated molecule is a viable drug candidate.
What benchmarking reveals:
- Strong on geometry and predicted binding. On standard structure-based benchmarks, leading diffusion models generate molecules with good 3D geometry and favorable predicted binding affinity, which is the result most often highlighted.
- Weak on chemical realism. A recurring finding is that structure-based diffusion models often produce unrealistic 3D molecules that neglect structural feasibility and drug-like properties, generating shapes that score well on binding but would not make sensible drugs.
- Benchmark dependence. Performance rankings shift depending on the benchmark and the metrics chosen, so comparisons across models must be read carefully, and headline claims of superiority often rest on a specific evaluation setup.
- Property guidance helps. Newer models explicitly incorporate binding affinity and drug-like property guidance into training and sampling, and these generate molecules with more rational chemical structure, a direct response to the realism problem.
The honest summary from the evaluation literature is that many structure-based diffusion models produce unrealistic molecules unless structural feasibility and drug-likeness are explicitly built in, and that addressing this, through bond modeling and property guidance, is where the field is actively improving. Impressive binding scores on a benchmark are necessary but not sufficient evidence that a model is useful for real drug design.
From generated structure to chemistry
The final and most important question is what happens when a diffusion-generated molecule leaves the screen and meets a medicinal chemist. Here, the same reality that applies across generative design applies with particular force: a 2025 review of diffusion models in drug discovery identifies the core challenge for small molecules as bridging the gap from computational validity to practical chemical synthesizability. A molecule that a diffusion model places beautifully inside a pocket is only useful if a chemist can actually make it.
What has to happen after generation:
- Synthesizability assessment. Generated molecules must be checked for whether they can realistically be synthesized, and a meaningful fraction of raw diffusion output fails this test, containing strained, reactive, or simply unmakeable substructures.
- Drug-likeness filtering. Beyond synthesis, generated molecules are screened against drug-like property criteria, since a high predicted binding score means little if the molecule could never be developed into a viable therapeutic.
- Experimental validation. A predicted binding pose is a hypothesis, not a fact, and confirming that a generated molecule actually binds as designed still requires experimental work, connecting back to the limits of prediction discussed for structure-based methods generally.
- Medicinal chemistry judgment. Deciding which generated candidates are worth pursuing remains an expert task, with diffusion output serving as a source of well-shaped starting points rather than finished designs.
The balanced conclusion is that diffusion models are a genuine advance in generating novel 3D molecules, particularly for structure-based design, and that their trajectory, toward built-in chemical realism, is encouraging. But the gap between a geometrically impressive generated structure and a synthesizable, developable drug remains the decisive one, and closing it is where the practical value of these models will ultimately be judged. For how this fits the wider design landscape, DDN's generative AI in molecular design hub provides the surrounding context.
What this means for drug design teamsDiffusion models are worth taking seriously for structure-based design, where generating molecules directly inside a target pocket in 3D is a real capability, not a demo. Use DiffDock-style models for pose prediction and DiffSBDD- or TargetDiff-style models for pocket-conditioned generation, but read benchmark claims critically: strong geometry and predicted binding do not guarantee chemically realistic molecules. Prioritize models and workflows that build synthesizability and drug-like property guidance into generation rather than filtering afterward, since raw diffusion output often includes molecules that cannot be made or developed. Treat generated structures as well-shaped starting points for medicinal chemistry, confirm binding experimentally, and keep expert judgment in the loop on what is worth pursuing. For the broader design picture, the generative AI in molecular design guide and the AI in drug discovery overview map how 3D generation connects to the rest of the pipeline. |
This article was produced in accordance with Drug Discovery News’ Editorial Policies.
















