arXiv Machine Learning
Trajectory Matching: Discrete Flow Matching with Structured Actions
Read the original on arXiv Machine Learning →The Flow has not summarised this story yet — read it at arXiv Machine Learning.
Related stories
arXiv AI
Aug 28
Mechanistic Reaction Prediction via Discrete Flow Matching on Graph-Structured Electron Occupation
MAELLE is a mechanistic reaction prediction framework that models chemical reactions as discrete flow matching over graph-structured electron occupation vectors. It formulates the reactant-to-product mapping as a Continuous-time Markov Chain on electron sites and uses Optimal Transport to generate mechanistically interpretable edit trajectories without elementary step annotations. The method achieves competitive accuracy on the USPTO-480K benchmark, remains robust in out-of-distribution scenarios, and can recover mechanistic pathways that align with known chemistry and predict side products.
arXiv Machine Learning
Jun 8
Controllable Molecular Generative Foundation Models
arXiv:2605. 15354v2 Announce Type: replace Abstract: Despite the success of foundation models in language and vision, molecular graph generation still lacks a unified framework for heterogeneous design tasks with reliable controllability.
arXiv AI
Aug 24
ReCurveflow: A Flow Matching Framework that Learns Curved Reaction Trajectories to Predict Transition State Geometries
ReCurveflow is a flow‑matching framework that learns to predict transition state geometries by training on continuously curved reference paths derived from full NEB bands, rather than straight linear paths. It introduces an off‑path correction mechanism that generates corrective velocity fields when the model encounters geometries off the training path, improving resistance to exposure bias and TS prediction accuracy. Across multiple data splits and evaluation metrics, ReCurveflow outperforms seven baselines and produces reaction trajectories whose energy profiles closely follow the reference NEB path, aiding NEB optimization and demonstrating effective corrective behavior.
arXiv Machine Learning
Jul 22
GEqTrain: A Configuration-Driven Framework for Retargeting Equivariant Graph Neural Networks Across 3D Scientific Tasks
arXiv:2607. 19083v1 Announce Type: new Abstract: Equivariant graph neural networks provide a powerful modeling language for three-dimensional scientific data, but their reuse is often limited by implementations tied to specific tasks, outputs, and training regimes.
arXiv Machine Learning
Aug 31
Trajectory balance: Improved credit assignment in GFlowNets
The paper introduces a new learning objective called trajectory balance for Generative Flow Networks (GFlowNets), aiming to improve credit assignment across long action sequences. It demonstrates that minimizing this objective yields a policy that samples exactly from the target distribution. Experiments on four domains show that trajectory balance enhances convergence, sample diversity, and robustness to long sequences and large action spaces.
arXiv AI
Aug 7
D-CLOT: Double Closed Loop Optimal Transport for Unsupervised Action Segmentation
arXiv:2608. 05877v1 Announce Type: cross Abstract: Optimal transport (OT) has emerged as an effective framework for unsupervised action segmentation.