CausticFlow: An Efficient Machine Learning Framework Combining Neural Differential Equations and Normalizing Flows for Binary Microlensing Parameter Inference
Abstract
We introduce CausticFlow, a machine learning framework that combines neural controlled differential equations with normalizing flows to infer binary microlensing parameters. This architecture naturally handles irregularly sampled time series and data gaps while flexibly capturing strongly correlated and multimodal posterior distributions. Trained on simulated KMTNet-like light curves, CausticFlow generates posterior samples in a fraction of a second, with maximum-a-posteriori estimates achieving typical precisions of ∼17% for the mass ratio q and ∼3% for the projected separation s. When used as a proposal distribution for downstream local optimization, the framework improves these precisions to <5% and <1%, respectively, and recovers model χ2 for ∼80% of simulated events. We test the generalizability of the framework on 10 real binary lensing events characterized by higher-order effects, varied cadences, and real-world noise. Despite these mismatches between simulation and reality, CausticFlow successfully recovers the model parameters, light-curve morphology, and lensing geometry for 7 of the 10 events after simple local refinement, achieving precision levels comparable to those found for simulated data in 10 CPU minutes per event. These results demonstrate that CausticFlow acts as a fast and robust proposal engine that provides informative starting points for downstream local optimization, bridging the gap between the rapid influx of data and the need for systematic modeling in large-scale microlensing surveys such as Roman, CSST, and ET.
Versions
Related
- Revisiting the Einstein Cross with X-shooter spectra: New estimates of accretion disk parameters and broad emission line profiles from microlensing
- Model-driven Stellar Spectroscopy with the Total Differential Spectral Model ( TDSM ): Framework and Its Application to 11 Million LAMOST Low-resolution Spectra
- lenscape: Fast, High-resolution Δχ2 Maps of Microlensing Events
- Same Events, Weaker Bound: Model Dependence in Magellanic Microlensing Limits on Primordial Black Holes
- Recovering Weak Signals with Normalizing Flows
BibTeX
@misc{ren2026causticflow,
title = {{CausticFlow: An Efficient Machine Learning Framework Combining Neural Differential Equations and Normalizing Flows for Binary Microlensing Parameter Inference}},
author = {Haibin Ren and Wei Zhu},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2607.04955},
eprint = {2607.04955},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2607.04955},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - CausticFlow: An Efficient Machine Learning Framework Combining Neural Differential Equations and Normalizing Flows for Binary Microlensing Parameter Inference
AU - Haibin Ren
AU - Wei Zhu
PY - 2026
DA - 2026/10/13
PB - arXiv
DO - 10.48550/arxiv.2607.04955
UR - https://arxiv.org/abs/2607.04955
AB - We introduce CausticFlow, a machine learning framework that combines neural controlled differential equations with normalizing flows to infer binary microlensing parameters. This architecture naturally handles irregularly sampled time series and data gaps while flexibly capturing strongly correlated and multimodal posterior distributions. Trained on simulated KMTNet-like light curves, CausticFlow generates posterior samples in a fraction of a second, with maximum-a-posteriori estimates achieving typical precisions of ∼17% for the mass ratio q and ∼3% for the projected separation s. When used as a proposal distribution for downstream local optimization, the framework improves these precisions to <5% and <1%, respectively, and recovers model χ^2 for ∼80% of simulated events. We test the generalizability of the framework on 10 real binary lensing events characterized by higher-order effects, varied cadences, and real-world noise. Despite these mismatches between simulation and reality, CausticFlow successfully recovers the model parameters, light-curve morphology, and lensing geometry for 7 of the 10 events after simple local refinement, achieving precision levels comparable to those found for simulated data in 10 CPU minutes per event. These results demonstrate that CausticFlow acts as a fast and robust proposal engine that provides informative starting points for downstream local optimization, bridging the gap between the rapid influx of data and the need for systematic modeling in large-scale microlensing surveys such as Roman, CSST, and ET.
ER -