AReUReDi: Annealed Rectified Updates for Refining Discrete Flows with Multi-Objective Guidance

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Tong, Zhang, Yinuo, Chatterjee, Pranam
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911190196682752
author Chen, Tong
Zhang, Yinuo
Chatterjee, Pranam
author_facet Chen, Tong
Zhang, Yinuo
Chatterjee, Pranam
contents Designing sequences that satisfy multiple, often conflicting, objectives is a central challenge in therapeutic and biomolecular engineering. Existing generative frameworks largely operate in continuous spaces with single-objective guidance, while discrete approaches lack guarantees for multi-objective Pareto optimality. We introduce AReUReDi (Annealed Rectified Updates for Refining Discrete Flows), a discrete optimization algorithm with theoretical guarantees of convergence to the Pareto front. Building on Rectified Discrete Flows (ReDi), AReUReDi combines Tchebycheff scalarization, locally balanced proposals, and annealed Metropolis-Hastings updates to bias sampling toward Pareto-optimal states while preserving distributional invariance. Applied to peptide and SMILES sequence design, AReUReDi simultaneously optimizes up to five therapeutic properties (including affinity, solubility, hemolysis, half-life, and non-fouling) and outperforms both evolutionary and diffusion-based baselines. These results establish AReUReDi as a powerful, sequence-based framework for multi-property biomolecule generation.
format Preprint
id arxiv_https___arxiv_org_abs_2510_00352
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AReUReDi: Annealed Rectified Updates for Refining Discrete Flows with Multi-Objective Guidance
Chen, Tong
Zhang, Yinuo
Chatterjee, Pranam
Machine Learning
Biomolecules
Designing sequences that satisfy multiple, often conflicting, objectives is a central challenge in therapeutic and biomolecular engineering. Existing generative frameworks largely operate in continuous spaces with single-objective guidance, while discrete approaches lack guarantees for multi-objective Pareto optimality. We introduce AReUReDi (Annealed Rectified Updates for Refining Discrete Flows), a discrete optimization algorithm with theoretical guarantees of convergence to the Pareto front. Building on Rectified Discrete Flows (ReDi), AReUReDi combines Tchebycheff scalarization, locally balanced proposals, and annealed Metropolis-Hastings updates to bias sampling toward Pareto-optimal states while preserving distributional invariance. Applied to peptide and SMILES sequence design, AReUReDi simultaneously optimizes up to five therapeutic properties (including affinity, solubility, hemolysis, half-life, and non-fouling) and outperforms both evolutionary and diffusion-based baselines. These results establish AReUReDi as a powerful, sequence-based framework for multi-property biomolecule generation.
title AReUReDi: Annealed Rectified Updates for Refining Discrete Flows with Multi-Objective Guidance
topic Machine Learning
Biomolecules
url https://arxiv.org/abs/2510.00352