REVIEW 5 major objections 5 minor 32 references
BoKDiff: Best-of-K Diffusion Alignment for Target-Specific 3D Molecule Generation
T0 review · 5 major / 5 minor · reviewed 2026-08-10 · deepseek-v4-flash
Pith's one-line read A best-of-K fine-tuning loop over a base diffusion model's own samples improves 3D molecule generation for drug targets, reaching -8.58 average Vina docking score and 26% success rate.
desk verdict Useful engineering recipe for aligning DecompDiff, but the evaluation protocol undermines the SOTA claim; the headline numbers are partly selected on the test set. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The central object is the DecompDiff generator, a diffusion model that produces atoms and bonds conditioned on data-dependent priors built from protein sub-pockets and ligand fragments (arms and a scaffold). Around it, BoKDiff wraps a Best-of-K reward-ranked fine-tuning loop (RAFT): sample K candidate ligands per pocket, score them with a weighted reward of QED, SA, and Vina dock, keep the top one, relocate its center of mass to the docking pose's center of mass, re-extract sub-pockets and substructures with adjusted Alphaspace2 parameters, and fine-tune the generator on the curated sample. The relocation is the paper's key adaptation; without it, generated ligands are often too far from the protein for substructure extraction to succeed.
What would settle it
A reader could test the mechanism by taking a fixed batch of generated ligands and comparing substructure-extraction success with and without the center-of-mass relocation; if extraction succeeds at similar rates without relocation, or fails often even with it, the alignment step is not doing the work attributed to it. The paper does not report extraction success rates, only the final aggregate metrics.
Extended reading notes
Core claim
The central claim is that a reward-ranked fine-tuning loop, adapted from RAFT, can push a strong diffusion generator beyond its original performance on binding and drug-likeness metrics. Concretely, BoKDiff samples K=10 ligands per training pocket, scores them with a weighted reward (with the best configuration found to be QED weight 1, SA weight 0, Vina weight 0), keeps the top-scoring ligand, relocates its center of mass to the center of mass of its Vina docking pose, and fine-tunes DecompDiff for 1,000 iterations at a learning rate of 1e-6 on the resulting curated set. The paper reports that this procedure raises mean Vina Dock from -8.39 to -8.58, success rate from 24.5% to 26.0%, and mean QED from 0.45 to 0.48 relative to DecompDiff, and it argues that the center-of-mass relocation is what makes the fine-tuning signal usable by enabling reliable sub-pocket and substructure extraction. It also claims that a no-fine-tuning Best-of-N sampling strategy, selecting the best of 20 generated samples by the same reward, achieves higher QED, SA, and success rate than the fine-tuned model, at greater inference cost.
Load-bearing premise
The data-preparation step assumes that translating a generated ligand so its center of mass coincides with that of its docking pose preserves the ligand's structural integrity and lets the modified Alphaspace2 extraction reliably identify sub-pockets and substructures; if that alignment corrupts or misassigns components, the fine-tuning signal is degraded and the gains would not generalize.
Editorial extensions
If this is right
- A base diffusion model for structure-based drug design can be improved on docking and success-rate metrics by fine-tuning on its own best-ranked generated samples, with no change to the model architecture and a fine-tuning cost of roughly 10 minutes on one GPU.
- Because the optimal reward configuration was QED-only, the paper implies that optimizing drug-likeness alone can transfer to improved Vina metrics and success rate for this model.
- The Best-of-N results indicate that simply drawing N=20 samples and picking the highest-reward one yields success rates above 35% and QED above 0.65, outperforming the fine-tuned model on those metrics at higher inference cost.
- The success of the pipeline depends on having a docking pose for each generated sample; every improvement in the paper is therefore conditional on the availability and quality of those poses.
Reading between the lines
- An untested extension is to relocate to multiple docking poses per ligand and aggregate the reward, which the paper lists as future work but does not run; if the relocation mechanism is what matters, pose diversity should sharpen the reward signal.
- The success-rate threshold comes from Jin et al. and is permissive (QED > 0.25, SA > 0.59, Vina Dock < -8.18); adopting a more stringent threshold could reorder the relative standing of BoKDiff and Best-of-N.
- The paper's comparison of CoM relocation to the docking pose versus the reference ligand suggests that docking-pose alignment is the more useful signal; a natural next probe is whether partial rotations toward the pose add further gains without breaking substructure extraction.
- The finding that a QED-only reward also lifts Vina metrics may be specific to DecompDiff's inductive bias (it favors simpler arms), so transferring BoKDiff to a different generator may not reproduce the same correlation.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes BoKDiff, a fine-tuning framework built on DecompDiff for target-specific 3D molecule generation. BoKDiff applies the RAFT reward-ranked fine-tuning recipe: for each protein in a batch, it generates K ligand samples, ranks them with a weighted sum of QED, SA, and Vina docking score, relocates the best sample's center of mass to that of its docking pose, extracts sub-pockets and substructures from the relocated sample, and fine-tunes the base diffusion model on the curated high-reward set. The paper also evaluates a Best-of-N sampling strategy that skips fine-tuning and directly selects the highest-reward candidate from N samples. The headline claims are a -8.58 average Vina docking score and a 26% success rate on CrossDocked2020, presented as state-of-the-art results, together with a BoN success rate above 35%. The main methodological novelty is the application of best-of-K alignment and BoN sampling to structure-based drug design, plus a center-of-mass relocation step intended to fix ligand-protein misalignment before substructure extraction.
Significance. If the evaluation were sound, this would be a useful and clearly described contribution: it transplants a successful LLM alignment strategy (RAFT) into structure-based drug design, provides a concrete fix for the misalignment problem in decomposed diffusion models, and includes a public code repository, which is valuable for reproducibility. The center-of-mass relocation idea is simple and potentially transferable to other SBDD pipelines. However, the current manuscript does not establish the state-of-the-art claim because the central evaluation protocol selects hyperparameters on the same test set used for final reporting, reports no uncertainty estimates, and relies on small aggregate differences over a single run. The core alignment mechanism may well be sound, but the evidence as presented is not yet sufficient for the claimed strength of the results.
major comments (5)
- [Weight Configurations, Tables 4-5] The optimal weight configuration [QED:1, SA:0, Vina:0] is selected by comparing five configurations on 30 samples taken from the test set (Tables 4 and 5), and the same configuration is then used to produce the headline results in Table 1. This is test-set selection: the hyperparameter that drives the -8.58 Vina Dock and 26.0% success rate is chosen by looking at the evaluation proteins. Moreover, the selection signal in Table 4 is very weak, with accumulated mean scores of 1.880 for [1,0,0] versus 1.867 for [1,1,1], a difference of 0.013, and no confidence intervals or significance tests are given. Please re-run the weight selection on a validation split that is disjoint from the reported test set, and report the variability of the selection criterion.
- [Ablation Studies, Tables 4-6] The ablation studies are described as 'conducted on 30 test set samples,' and the table captions state that the results use '30 data samples selected from the test set.' Since the BoKDiff pipeline fine-tunes on the collected samples, this means the model is fine-tuned and evaluated on the same test proteins. This contamination undermines the conclusions about weight configurations, batch size, iteration number, and the docking-pose versus reference-ligand relocation comparison. Please use a separate validation set for all ablations and report results on a truly held-out test set.
- [Main Results, Table 1] Table 1 reports a single mean and median per metric with no error bars, no per-protein paired analysis, and no statement about the number of random seeds or independent runs. The differences that support the state-of-the-art claim are small: BoKDiff versus DecompDiff gives Vina Dock -8.58 versus -8.39 and success rate 26.0% versus 24.5%. With roughly 1000 generated molecules, these differences may be within sampling noise, especially if docking seeds and generation randomness are not controlled. Please report repeated runs, per-protein paired comparisons, and confidence intervals, and clarify whether the baseline numbers in Table 1 were re-run under identical conditions or taken from prior publications.
- [Data Preparation] The center-of-mass relocation step is central to the method, but the paper does not directly validate its key assumption: that translating a generated ligand so that its center of mass coincides with the center of mass of its Vina docking pose preserves the ligand's structural integrity and allows Alphaspace2 to extract the correct sub-pockets and substructures. The initial attempt with raw docking poses is said to alter ligand structure, and the proposed fix avoids that alteration, yet no quantitative evidence is given that the relocated ligand still corresponds to a valid, extractable decomposition. Table 6 compares end metrics for two relocation sources, but it does not show extraction success rates, substructure assignment fidelity, or validity before and after relocation. Please add direct validation of the extraction step.
- [Best-of-N, Table 3] The BoN results in Table 3 are also affected by selection on the evaluation data: the text states that the weight set [1.1,1,0.9] 'was identified through a grid search centered around the baseline [1,1,1],' and then Table 3 reports performance for that selected configuration. The claim that BoN surpasses 35% success rate is therefore based on the best configuration found on the same data used to report the result. In addition, the sentence 'other combinations emphasizing SA or Vina Dock are omitted due to their similarity to [1,0,0]' is inconsistent with the table, which includes [1,1,1] and [1.1,1,0.9]. Please clarify the selection procedure and report BoN results on a held-out set.
minor comments (5)
- [Title page] The manuscript still contains placeholder header text ('Journal Title Here', 'FOR PUBLISHER ONLY', '© The Author 2022') and several typographical issues such as 'F or permissions' and 'Y ear'; these should be cleaned before submission.
- [Data Ranking section] The description of Vina score normalization is underspecified: min-max scaling across the K=10 samples per input depends on the particular set of generated samples and can be unstable when K is small; please state the exact formula and how ties are handled.
- [Equations (1)-(3)] The notation for the DecompDiff equations is not fully self-contained: the variables H, Σ, µ, and the tilde notation for prior-centered positions are introduced only briefly. Readers not already familiar with DecompDiff would benefit from explicit definitions.
- [Best-of-N section] The abstract and Section 5 both claim QED values exceeding 0.6, SA above 0.75, and success above 35% for BoN, but these numbers only hold for the larger N values and for selected weight sets; the text should make clear that these are properties of a particular, data-dependent configuration rather than of BoN sampling per se.
- [Experimental Setup] The implementation details state that fine-tuning is run for '1,000 epochs' but earlier text says '1,000 iterations' and the original model is trained for 500,000 iterations; please use consistent terminology and specify what one epoch or iteration means.
Circularity Check
BoN's headline property results are its own selection objective, and the BoKDiff reward weights were tuned on a subset of the same test set used for the reported SOTA numbers.
-
self definitional
[Best-of-N section and Table 3; Data Ranking (Practical Implementation)]
"Instead of curating a high-reward dataset by selecting the best-of-K samples and fine-tuning the model based on this dataset, an alternative approach skips the fine-tuning step entirely by directly selecting the best sample from N generated samples. ... the selection of the best sample relies entirely on its reward value. [Table 3 caption:] Comparison of Best-of-N performance on the reference model for various values of N . Results are provided for the weight combination prioritizing QED, while other combinations emphasizing SA or Vina Dock are omitted due to their similarity to [1, 0, 0]."
The reward used for ranking is explicitly a weighted sum of QED, SA, and Vina Dock. Best-of-N is defined as choosing the candidate that maximizes this reward, and Table 3 then reports QED, SA, and success rate (a threshold over QED/SA/Vina Dock) for the chosen candidate. The improvement of these metrics with N is an order-statistic consequence of maximizing the very objective being reported; no generative model is changed and no independent benchmark is used. The 'exceptional results' in the abstract are therefore the selection criterion by construction, not an external validation of the generative method.
-
fitted input called prediction
[Ablation Studies, Weight Configurations (Tables 4-5) and Main Results, Table 1]
"These results correspond to the first iteration, using 30 data samples selected from the test set. (Accumulated columns represent the simple summation (with equal weights) of three metrics, where the Vina Dock scores are normalized by dividing them by -10 to ensure they fall within the [0, 1] range.) ... Notably, the best-performing weight configuration was (QED:1, SA:0, Vina Dock:0). [Table 1:] This experiment was conducted using our optimal weight configuration (QED: 1, SA: 0, Vina: 0)."
The reward-weight hyperparameter controls which generated molecules are selected into the fine-tuning set. It was chosen by comparing accumulated QED/SA/Vina scores on 30 proteins explicitly drawn from the test set, and the headline Table 1 results are then reported on the same test set under that 'optimal' configuration. The configuration is thus a fitted parameter selected from the evaluation data, and the reported state-of-the-art numbers are not an independent out-of-sample prediction; they are the outcome of choosing the hyperparameter that looked best on a subset of the benchmark. The remaining 70 test proteins are not reported separately, so the claimed gain is partly in-sample.
full rationale
The paper's core engineering contribution — CoM relocation plus RAFT-style best-of-K fine-tuning of DecompDiff — is not circular: DecompDiff, RAFT, and BoN are external methods, and the fine-tuned model is assessed with standard affinity and property metrics. There is no self-citation chain and no imported uniqueness theorem. However, two load-bearing numerical claims reduce to their own inputs. First, the abstract's BoN 'exceptional results' (QED>0.6, SA>0.75, success>35%) are reported on the candidate that BoN selects by maximizing a reward that is literally a weighted sum of QED, SA, and Vina Dock; as N increases these numbers improve by the order statistics of the selection objective, so the prediction is the selection criterion by construction. Second, the weight configuration used for the Table 1 BoKDiff state-of-the-art claim was chosen as 'best-performing' from ablations run on 30 proteins drawn from the same test set; the final results are therefore a hyperparameter fit to a subset of the benchmark rather than a held-out evaluation. Neither issue involves self-citation, but both affect the paper's headline quantitative claims. The CoM-relocation and fine-tuning pipeline itself is non-circular and could be validated by a properly held-out protocol.
Assumptions & free parameters
free parameters (7)
- Reward weights [QED, SA, Vina Dock] =
[1, 0, 0] for BoKDiff; [1.1, 1, 0.9] for BoN, selected by grid search
- K (number of candidates per protein in data collection) =
10
- Batch size B for high-reward dataset =
128
- Fine-tuning steps =
1000 (called both iterations and epochs)
- Learning rate =
1e-6
- N for Best-of-N sampling =
20
- Cutoff and Radius for Alphaspace2 extraction =
Minimum values determined dynamically, exact values not reported
assumptions (4)
- domain assumption DecompDiff generates valid, diverse ligand candidates from decomposed structural priors.
- domain assumption QED, SA, and Vina docking score are reliable computational proxies for drug-likeness, synthesizability, and binding affinity.
- domain assumption The generative model can achieve the maximum reward for each input protein, so per-input best-of-K selection followed by SFT maximizes expected reward.
- domain assumption Vina docking poses provide a dependable reference for relocating generated ligands near the binding site.
Cite this review
Pith. "Pith review of BoKDiff: Best-of-K Diffusion Alignment for Target-Specific 3D Molecule Generation." pith.science (2026). https://pith.science/paper/EUUYB4Q6
@misc{pith2026250115631,
author = {Pith},
title = {Pith review of: BoKDiff: Best-of-K Diffusion Alignment for Target-Specific 3D Molecule Generation},
year = {2026},
howpublished = {\url{https://pith.science/paper/EUUYB4Q6}},
note = {Machine review of arXiv:2501.15631}
}
read the original abstract
Structure-based drug design (SBDD) leverages the 3D structure of biomolecular targets to guide the creation of new therapeutic agents. Recent advances in generative models, including diffusion models and geometric deep learning, have demonstrated promise in optimizing ligand generation. However, the scarcity of high-quality protein-ligand complex data and the inherent challenges in aligning generated ligands with target proteins limit the effectiveness of these methods. We propose BoKDiff, a novel framework that enhances ligand generation by combining multi-objective optimization and Best-of-K alignment methodologies. Built upon the DecompDiff model, BoKDiff generates diverse candidates and ranks them using a weighted evaluation of molecular properties such as QED, SA, and docking scores. To address alignment challenges, we introduce a method that relocates the center of mass of generated ligands to their docking poses, enabling accurate sub-component extraction. Additionally, we integrate a Best-of-N (BoN) sampling approach, which selects the optimal ligand from multiple generated candidates without requiring fine-tuning. BoN achieves exceptional results, with QED values exceeding 0.6, SA scores above 0.75, and a success rate surpassing 35%, demonstrating its efficiency and practicality. BoKDiff achieves state-of-the-art results on the CrossDocked2020 dataset, including a -8.58 average Vina docking score and a 26% success rate in molecule generation. This study is the first to apply Best-of-K alignment and Best-of-N sampling to SBDD, highlighting their potential to bridge generative modeling with practical drug discovery requirements. The code is provided at https://github.com/khodabandeh-ali/BoKDiff.git.
Figures
Reference graph
Works this paper leans on
-
[1]
The process of structure-based drug design
Amy C Anderson. The process of structure-based drug design. Chemistry & biology , 10(9):787–797, 2003
work page 2003
-
[2]
Quantifying the chemical beauty of drugs
G Richard Bickerton, Gaia V Paolini, J´ er´ emy Besnard, Sorel Muresan, and Andrew L Hopkins. Quantifying the chemical beauty of drugs. Nature chemistry, 4(2):90–98, 2012
work page 2012
-
[3]
T raining diffusion models with reinforcement BoKDiff 9 learning
Kevin Black, Michael Janner, Yilun Du, Ilya Kostrikov, and Sergey Levine. T raining diffusion models with reinforcement BoKDiff 9 learning. arXiv preprint arXiv:2305.13301 , 2023
arXiv 2023
-
[4]
Deep reinforcement learning from human preferences
Paul F Christiano, Jan Leike, Tom Brown, Miljan Martic, Shane Legg, and Dario Amodei. Deep reinforcement learning from human preferences. Advances in neural information processing systems, 30, 2017
2017
-
[5]
Raft: Reward ranked finetuning for generative foundation model alignment
Hanze Dong, Wei Xiong, Deepanshu Goyal, Yihan Zhang, Winnie Chow, Rui Pan, Shizhe Diao, Jipeng Zhang, Kashun Shum, and Tong Zhang. Raft: Reward ranked finetuning for generative foundation model alignment. arXiv preprint arXiv:2304.06767, 2023
arXiv 2023
-
[6]
Jerome Eberhardt, Diogo Santos-Martins, Andreas F Tillack, and Stefano Forli. Autodock vina 1.2. 0: New docking methods, expanded force field, and python bindings. Journal of chemical information and modeling , 61(8):3891–3898, 2021
work page 2021
-
[7]
Peter Ertl and Ansgar Schuffenhauer. Estimation of synthetic accessibility score of drug-like molecules based on molecular complexity and fragment contributions. Journal of cheminformatics , 1:1–11, 2009
work page 2009
-
[8]
Paul G Francoeur, Tomohide Masuda, Jocelyn Sunseri, Andrew Jia, Richard B Iovanisci, Ian Snyder, and David R Koes. Three-dimensional convolutional neural networks and a cross-docked data set for structure-based drug design. Journal of chemical information and modeling , 60(9):4200–4215, 2020
work page 2020
Show all 32 references
-
[9]
Reinforced genetic algorithm for structure-based drug design
Tianfan Fu, Wenhao Gao, Connor Coley, and Jimeng Sun. Reinforced genetic algorithm for structure-based drug design. Advances in Neural Information Processing Systems, 35:12325–12338, 2022
2022
-
[10]
3d equivariant diffusion for target-aware molecule generation and affinity prediction
Jiaqi Guan, Wesley Wei Qian, Xingang Peng, Yufeng Su, Jian Peng, and Jianzhu Ma. 3d equivariant diffusion for target-aware molecule generation and affinity prediction. arXiv preprint arXiv:2303.03543 , 2023
2023 arXiv
-
[11]
Decompdiff: diffusion models with decomposed priors for structure-based drug design
Jiaqi Guan, Xiangxin Zhou, Yuwei Yang, Yu Bao, Jian Peng, Jianzhu Ma, Qiang Liu, Liang Wang, and Quanquan Gu. Decompdiff: diffusion models with decomposed priors for structure-based drug design. arXiv preprint arXiv:2403.07902, 2024
2024 arXiv
-
[12]
Bonbon alignment for large language models and the sweetness of best-of-n sampling
Lin Gui, Cristina Gˆ arbacea, and Victor Veitch. Bonbon alignment for large language models and the sweetness of best-of-n sampling. arXiv preprint arXiv:2406.00832 , 2024
2024 arXiv
-
[13]
Multi-objective molecule generation using interpretable substructures
Wengong Jin, Regina Barzilay, and Tommi Jaakkola. Multi-objective molecule generation using interpretable substructures. In International conference on machine learning, pages 4849–4859. PMLR, 2020
2020
-
[14]
Diffbp: Generative diffusion of 3d molecules for target protein binding
Haitao Lin, Yufei Huang, Meng Liu, Xuanjing Li, Shuiwang Ji, and Stan Z Li. Diffbp: Generative diffusion of 3d molecules for target protein binding. arXiv preprint arXiv:2211.11214, 2022
2022 arXiv
-
[15]
Generating 3d molecules for target protein binding
Meng Liu, Youzhi Luo, Kanji Uchino, Koji Maruhashi, and Shuiwang Ji. Generating 3d molecules for target protein binding. arXiv preprint arXiv:2204.09410 , 2022
2022 arXiv
-
[16]
A 3d generative model for structure-based drug design
Shitong Luo, Jiaqi Guan, Jianzhu Ma, and Jian Peng. A 3d generative model for structure-based drug design. Advances in Neural Information Processing Systems , 34:6229–6239, 2021
2021
-
[17]
Training language models to follow instructions with human feedback
Long Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida, Carroll Wainwright, Pamela Mishkin, Chong Zhang, Sandhini Agarwal, Katarina Slama, Alex Ray, et al. Training language models to follow instructions with human feedback. Advances in neural information processing systems, 35:2773...
2022
-
[18]
Pocket2mol: Efficient molecular sampling based on 3d protein pockets
Xingang Peng, Shitong Luo, Jiaqi Guan, Qi Xie, Jian Peng, and Jianzhu Ma. Pocket2mol: Efficient molecular sampling based on 3d protein pockets. In International Conference on Machine Learning , pages 17644–17655. PMLR, 2022
2022
-
[19]
Geometric deep learning for structure-based ligand design
Alexander S Powers, Helen H Yu, Patricia Suriana, Rohan V Koodli, Tianyu Lu, Joseph M Paggi, and Ron O Dror. Geometric deep learning for structure-based ligand design. ACS Central Science, 9(12):2257–2267, 2023
2023
-
[20]
Direct preference optimization: Your language model is secretly a reward model
Rafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D Manning, Stefano Ermon, and Chelsea Finn. Direct preference optimization: Your language model is secretly a reward model. Advances in Neural Information Processing Systems, 36, 2024
2024
-
[21]
Generating 3d molecules conditional on receptor binding sites with deep generative models
Matthew Ragoza, Tomohide Masuda, and David Ryan Koes. Generating 3d molecules conditional on receptor binding sites with deep generative models. Chemical science, 13(9):2701–2713, 2022
2022
-
[22]
Structure-based drug design with equivariant diffusion models
Arne Schneuing, Yuanqi Du, Charles Harris, Arian Jamasb, Ilia Igashov, Weitao Du, Tom Blundell, Pietro Li´ o, Carla Gomes, Max Welling, et al. Structure-based drug design with equivariant diffusion models. arXiv preprint arXiv:2210.13695, 2022
-
[23]
Proximal policy optimization algorithms
John Schulman, Filip Wolski, Prafulla Dhariwal, Alec Radford, and Oleg Klimov. Proximal policy optimization algorithms. arXiv preprint arXiv:1707.06347 , 2017
2017 arXiv
-
[24]
Autogrow4: an open- source genetic algorithm for de novo drug design and lead optimization
Jacob O Spiegel and Jacob D Durrant. Autogrow4: an open- source genetic algorithm for de novo drug design and lead optimization. Journal of cheminformatics , 12:1–16, 2020
2020
-
[25]
Applications of machine learning in drug discovery and development
Jessica Vamathevan, Dominic Clark, Paul Czodrowski, Ian Dunham, Edgardo Ferran, George Lee, Bin Li, Anant Madabhushi, Parantu Shah, Michaela Spitzer, et al. Applications of machine learning in drug discovery and development. Nature reviews Drug discovery , 18(6):463– 477, 2019
2019
-
[26]
Diffusion model alignment using direct preference optimization
Bram Wallace, Meihua Dang, Rafael Rafailov, Linqi Zhou, Aaron Lou, Senthil Purushwalkam, Stefano Ermon, Caiming Xiong, Shafiq Joty, and Nikhil Naik. Diffusion model alignment using direct preference optimization. In Proceedings of the IEEE/CVF Conference on Computer Vision and...
2024
-
[27]
Large-scale reinforcement learning for diffusion models
Yinan Zhang, Eric Tzeng, Yilun Du, and Dmitry Kislyuk. Large-scale reinforcement learning for diffusion models. arXiv preprint arXiv:2401.12244 , 4, 2024
2024 arXiv
-
[28]
Learning subpocket prototypes for generalizable structure-based drug design
Zaixi Zhang and Qi Liu. Learning subpocket prototypes for generalizable structure-based drug design. In International Conference on Machine Learning , pages 41382–41398. PMLR, 2023
2023
-
[29]
Molecule generation for target protein binding with structural motifs
Zaixi Zhang, Yaosen Min, Shuxin Zheng, and Qi Liu. Molecule generation for target protein binding with structural motifs. In The Eleventh International Conference on Learning Representations, 2023
2023
-
[30]
Decompopt: Controllable and decomposed diffusion models for structure-based molecular optimization
Xiangxin Zhou, Xiwei Cheng, Yuwei Yang, Yu Bao, Liang Wang, and Quanquan Gu. Decompopt: Controllable and decomposed diffusion models for structure-based molecular optimization. arXiv preprint arXiv:2403.13829 , 2024
2024 arXiv
-
[31]
Antigen-specific antibody design via direct energy-based preference optimization
Xiangxin Zhou, Dongyu Xue, Ruizhe Chen, Zaixiang Zheng, Liang Wang, and Quanquan Gu. Antigen-specific antibody design via direct energy-based preference optimization. arXiv preprint arXiv:2403.16576 , 2024
2024 arXiv
-
[32]
Fine-tuning language models from human preferences
Daniel M Ziegler, Nisan Stiennon, Jeffrey Wu, Tom B Brown, Alec Radford, Dario Amodei, Paul Christiano, and Geoffrey Irving. Fine-tuning language models from human preferences. arXiv preprint arXiv:1909.08593 , 2019
1909 arXiv
Reviewed August 10, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.