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.

Ссылки и действия