ROOT: Rethinking Offline Optimization as Distributional Translation via Probabilistic Bridge

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dao, Manh Cuong, Tran, The Hung, Nguyen, Phi Le, Truong, Thao Nguyen, Hoang, Trong Nghia
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912665548357632
author Dao, Manh Cuong
Tran, The Hung
Nguyen, Phi Le
Truong, Thao Nguyen
Hoang, Trong Nghia
author_facet Dao, Manh Cuong
Tran, The Hung
Nguyen, Phi Le
Truong, Thao Nguyen
Hoang, Trong Nghia
contents This paper studies the black-box optimization task which aims to find the maxima of a black-box function using a static set of its observed input-output pairs. This is often achieved via learning and optimizing a surrogate function with that offline data. Alternatively, it can also be framed as an inverse modeling task that maps a desired performance to potential input candidates that achieve it. Both approaches are constrained by the limited amount of offline data. To mitigate this limitation, we introduce a new perspective that casts offline optimization as a distributional translation task. This is formulated as learning a probabilistic bridge transforming an implicit distribution of low-value inputs (i.e., offline data) into another distribution of high-value inputs (i.e., solution candidates). Such probabilistic bridge can be learned using low- and high-value inputs sampled from synthetic functions that resemble the target function. These synthetic functions are constructed as the mean posterior of multiple Gaussian processes fitted with different parameterizations on the offline data, alleviating the data bottleneck. The proposed approach is evaluated on an extensive benchmark comprising most recent methods, demonstrating significant improvement and establishing a new state-of-the-art performance. Our code is publicly available at https://github.com/cuong-dm/ROOT.
format Preprint
id arxiv_https___arxiv_org_abs_2509_16300
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ROOT: Rethinking Offline Optimization as Distributional Translation via Probabilistic Bridge
Dao, Manh Cuong
Tran, The Hung
Nguyen, Phi Le
Truong, Thao Nguyen
Hoang, Trong Nghia
Machine Learning
This paper studies the black-box optimization task which aims to find the maxima of a black-box function using a static set of its observed input-output pairs. This is often achieved via learning and optimizing a surrogate function with that offline data. Alternatively, it can also be framed as an inverse modeling task that maps a desired performance to potential input candidates that achieve it. Both approaches are constrained by the limited amount of offline data. To mitigate this limitation, we introduce a new perspective that casts offline optimization as a distributional translation task. This is formulated as learning a probabilistic bridge transforming an implicit distribution of low-value inputs (i.e., offline data) into another distribution of high-value inputs (i.e., solution candidates). Such probabilistic bridge can be learned using low- and high-value inputs sampled from synthetic functions that resemble the target function. These synthetic functions are constructed as the mean posterior of multiple Gaussian processes fitted with different parameterizations on the offline data, alleviating the data bottleneck. The proposed approach is evaluated on an extensive benchmark comprising most recent methods, demonstrating significant improvement and establishing a new state-of-the-art performance. Our code is publicly available at https://github.com/cuong-dm/ROOT.
title ROOT: Rethinking Offline Optimization as Distributional Translation via Probabilistic Bridge
topic Machine Learning
url https://arxiv.org/abs/2509.16300