arXiv · 2510.09784
Combined Representation and Generation with Diffusive State Predictive Information Bottleneck
Abstract
Generative modeling becomes increasingly data-intensive in high-dimensional spaces. In molecular science, where data collection is expensive and important events are rare, compression to lower-dimensional manifolds is especially important for various downstream tasks, including generation. We combine a time-lagged information bottleneck designed to characterize molecular important representations and a diffusion model in one joint training objective. The resulting protocol, which we term Diffusive State Predictive Information Bottleneck (D-SPIB), enables the balancing of representation learning and generation aims in one flexible architecture. Additionally, the model is capable of combining temperature information from different molecular simulation trajectories to learn a coherent and useful internal representation of thermodynamics. We benchmark D-SPIB on multiple molecular tasks and showcase its potential for exploring physical conditions outside the training set.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Richard John, Yunrui Qiu, Lukas Herron, Pratyush Tiwary. 2025-10-10. Combined Representation and Generation with Diffusive State Predictive Information Bottleneck. https://arxiv.org/abs/2510.09784
Cite the original work for its findings. Save a collection to share your selection of sources.