Provable Mixed-Noise Learning with Flow-Matching
2508.18122v1
cs.LG, math.OC
2025-08-27
Авторы:
Paul Hagemann, Robert Gruhlke, Bernhard Stankewitz, Claudia Schillings, Gabriele Steidl
Резюме на русском
#### Контекст
В данном исследовании рассматриваются смешанные шумы в байесовских задачах обратного уравнения, характеризуемые сложным структурным характером, таких как сумма аддитивных и мультипликативных компонент гауссовского шума. Обычно, методы интерпретации таких задач предполагают знание характеристик шума, что не всегда соответствует реальным условиям, особенно в прикладных задачах физики и химии. В настоящей работе авторы целью делают проектирование новых методов, которые позволят обрабатывать ситуации с неизвестным и разнородным шумом. Используя подходы, основанные на flow-based generative modeling, предлагается разработать инференциальный фреймворк, который бы способствовал моделированию параметров шума и оценке параметров модели.
#### Метод
Методология построена на формализме Expectation-Maximization (EM), в котором включена новая концепция flow-matching в качестве модели для generative modeling. Основной идеей является использование нейросетевого моделирования в E-шаге, дающее возможность оценивать шум и параметры модели с помощью симуляционно-бесплатных ODE-потоков. При этом в формулировке считается, что множество выборок шума генерируется с помощью стохастического процесса, описываемого параметрами, которые необходимо оценить. Алгоритм построен так, чтобы в краткосрочной траектории обеспечивал интерпретируемую оценку параметров в границах оптимальных моделей. Теоретические результаты включают доказательство того, что в случае бесконечного объема данных, EM-алгоритм сходится к тем правдивым параметрам шума, которые предполагаются в задаче.
#### Результаты
Проводились эксперименты на синтетических данных, нацеленные на оценку точности оценок шума и параметров модели. Использовались шумы с разнообразными структурными характеристиками, включая смешанные компоненты гауссовского процесса. Результаты показали, что способ, основанный на flow-matching, дает более точные оценки в сравнении с традиционными методами. Были проведены измерения качества интерпретации шума, что демонстрирует высокую эффективность применения данного подхода в задачах с смешанным шумом.
#### Значимость
Предложенный подход имеет широкие применения в области научных исследований, где требуется обработка данных с неизвестным шумом, включая моделирование физических и химических процессов. Основное преимущество заключается в увеличении точности и стабильности результатов благодаря интеграции flow-matching в EM-инференциальный фреймворк. Это позволяет решать задачи, где традиционные методы были неэффективны. Бу
Abstract
We study Bayesian inverse problems with mixed noise, modeled as a combination
of additive and multiplicative Gaussian components. While traditional inference
methods often assume fixed or known noise characteristics, real-world
applications, particularly in physics and chemistry, frequently involve noise
with unknown and heterogeneous structure. Motivated by recent advances in
flow-based generative modeling, we propose a novel inference framework based on
conditional flow matching embedded within an Expectation-Maximization (EM)
algorithm to jointly estimate posterior samplers and noise parameters. To
enable high-dimensional inference and improve scalability, we use
simulation-free ODE-based flow matching as the generative model in the E-step
of the EM algorithm. We prove that, under suitable assumptions, the EM updates
converge to the true noise parameters in the population limit of infinite
observations. Our numerical results illustrate the effectiveness of combining
EM inference with flow matching for mixed-noise Bayesian inverse problems.
Ссылки и действия
Дополнительные ресурсы: