Diversity-Driven Offline Multi-Objective Optimization via Nested Pareto Set Learning
Abstract
Lay Summary
When designing a drug molecule or a neural network, engineers must balance multiple conflicting goals, for instance, high efficacy versus low toxicity. In practice, testing each candidate design is extremely expensive, so we can only use historical data from past experiments. The challenge is that models trained on this limited data become unreliable when evaluating new, untested designs; they tend to make overly optimistic predictions about regions far from the training data. This causes optimization algorithms to pick "fake winners" that look good on paper but fail in reality, while many genuinely promising designs get overlooked. We developed DOMOO, a method that learns to generate diverse and reliable candidate designs from historical data alone, without new experiments. DOMOO first trains an "energy model" that flags how risky each design is, like a quality inspector on an assembly line. It then learns to map different trade-off preferences (e.g., "efficacy matters more than toxicity") to specific designs, while using the risk signal to steer away from unreliable regions. Finally, it selects a well-spread set of candidates, ensuring diverse options rather than a cluster of similar-looking ones. Across 40+ benchmark tasks spanning molecular design, robotics control, and neural architecture search, DOMOO achieves the best overall ranking compared to existing methods. By explicitly avoiding unreliable predictions, our method produces solutions that are not only diverse but also trustworthy, reducing the chance that a recommended design fails in the real world. This work highlights the importance of risk awareness in data-driven optimization for domains where every experiment counts.