% \def\isarxiv{1} %%% for uai 2025 camera-ready version, we comment this line

\ifdefined\isarxiv
\documentclass[11pt]{article}

\usepackage[numbers]{natbib}


\else

\documentclass[accepted]{uai2025} 
% \usepackage[american]{babel}
% \usepackage{natbib}
% \usepackage{mathtools}

\fi


\usepackage{amsmath}
\usepackage{amsthm}
\usepackage{amssymb}
\usepackage{algorithm}
\usepackage{subfig}
\usepackage{algpseudocode}
\usepackage{graphicx}
\usepackage{grffile}
\usepackage{wrapfig,epsfig}
\usepackage{url}
\usepackage{xcolor}
\usepackage{epstopdf}


\usepackage{bbm}
\usepackage{dsfont}

\usepackage[american]{babel}
\usepackage{natbib}
\bibliographystyle{plainnat}
    \renewcommand{\bibsection}{\subsubsection*{References}}
\usepackage{mathtools}

 
 
\allowdisplaybreaks
 

\ifdefined\isarxiv

\let\C\relax
\usepackage{tikz}
\usepackage{hyperref}  %%% arxiv don't allow this.
\hypersetup{colorlinks=true,citecolor=blue,linkcolor=blue} %%% Zhao : maybe we should comment this in submission.
\usetikzlibrary{arrows}
\usepackage[margin=1in]{geometry}

\else
% \usepackage{microtype}
\usepackage{hyperref}
% \definecolor{mydarkblue}{rgb}{0,0.08,0.45}
% \hypersetup{colorlinks=true, citecolor=mydarkblue,linkcolor=mydarkblue}
 

\fi


\graphicspath{{./figs/}}

\theoremstyle{plain}
\newtheorem{theorem}{Theorem}[section]
\newtheorem{lemma}[theorem]{Lemma}
\newtheorem{definition}[theorem]{Definition}
\newtheorem{notation}[theorem]{Notation}
%\newtheorem{proof}[theorem]{Proof}
\newtheorem{proposition}[theorem]{Proposition}
\newtheorem{corollary}[theorem]{Corollary}
\newtheorem{conjecture}[theorem]{Conjecture}
\newtheorem{assumption}[theorem]{Assumption}
\newtheorem{observation}[theorem]{Observation}
\newtheorem{fact}[theorem]{Fact}
\newtheorem{remark}[theorem]{Remark}
\newtheorem{claim}[theorem]{Claim}
\newtheorem{example}[theorem]{Example}
\newtheorem{problem}[theorem]{Problem}
\newtheorem{open}[theorem]{Open Problem}
\newtheorem{property}[theorem]{Property}
\newtheorem{hypothesis}[theorem]{Hypothesis}

\newcommand{\wh}{\widehat}
\newcommand{\wt}{\widetilde}
\newcommand{\ov}{\overline}
\newcommand{\N}{\mathcal{N}}
\newcommand{\R}{\mathbb{R}}
\newcommand{\F}{\mathcal{F}}
\newcommand{\G}{\mathcal{G}}
\newcommand{\True}{\mathrm{true}}
\newcommand{\Noisy}{\mathrm{noisy}}
\newcommand{\Clean}{\mathrm{clean}}
\newcommand{\Est}{\mathrm{est}}
\newcommand{\RHS}{\mathrm{RHS}}
\newcommand{\LHS}{\mathrm{LHS}}
\renewcommand{\d}{\mathrm{d}}
\renewcommand{\i}{\mathbf{i}}
\renewcommand{\tilde}{\wt}
\renewcommand{\hat}{\wh}
\newcommand{\Tmat}{{\cal T}_{\mathrm{mat}}}

\DeclareMathOperator*{\E}{{\mathbb{E}}}
\DeclareMathOperator*{\var}{\mathrm{Var}}
\DeclareMathOperator*{\Z}{\mathbb{Z}}
\DeclareMathOperator*{\C}{\mathbb{C}}
\DeclareMathOperator*{\D}{\mathcal{D}}
\DeclareMathOperator*{\median}{median}
\DeclareMathOperator*{\mean}{mean}
\DeclareMathOperator{\OPT}{OPT}
\DeclareMathOperator{\supp}{supp}
\DeclareMathOperator{\poly}{poly}

\DeclareMathOperator{\nnz}{nnz}
\DeclareMathOperator{\sparsity}{sparsity}
\DeclareMathOperator{\rank}{rank}
\DeclareMathOperator{\diag}{diag}
\DeclareMathOperator{\dist}{dist}
\DeclareMathOperator{\cost}{cost}
\DeclareMathOperator{\vect}{vec}
\DeclareMathOperator{\tr}{tr}
\DeclareMathOperator{\dis}{dis}
\DeclareMathOperator{\cts}{cts}



\makeatletter
\newcommand*{\RN}[1]{\expandafter\@slowromancap\romannumeral #1@}
\makeatother
% \newcommand{\Zhao}[1]{{\color{red}[Zhao: #1]}} 
%%%Change to intern name

\usepackage{lineno}
\def\linenumberfont{\normalfont\small}


\begin{document}

\ifdefined\isarxiv

\date{}



\else

\title{NRFlow: Towards Noise-Robust Generative Modeling \\ via High-Order Flow Matching}

\author[1]{Bo Chen}
\author[2]{Chengyue Gong}
\author[3]{Xiaoyu Li}
\author[4,$\dagger$]{Yingyu Liang}
\author[2]{Zhizhou Sha}
\author[4]{Zhenmei Shi}
\author[5,*]{Zhao Song}
\author[1]{Mingda Wan}
\author[6]{Xugang Ye}

 
\affil[1]{%
    Middle Tennessee State University, \textsuperscript{2}University of Texas at Austin, \textsuperscript{3}University of New South Wales
}
\affil[4]{%
    University of Wisconsin-Madison, \textsuperscript{5}University of California, Berkeley, \textsuperscript{6}TikTok
}
\affil[*]{
    \texttt{magic.linuxkde@gmail.com}, \textsuperscript{$\dagger$}~\texttt{yliang@cs.wisc.edu,yingyul@hku.hk}
}
 

\fi





\ifdefined\isarxiv
\begin{titlepage}
  \maketitle
  \begin{abstract}
\input{0_abstract}

  \end{abstract}
  \thispagestyle{empty}
\end{titlepage}

{\hypersetup{linkcolor=black}
\tableofcontents
}
\newpage

\else
\maketitle
\begin{abstract}
\input{0_abstract}
\end{abstract}

\fi


\input{1_intro} %%% Section 1. Introduction
\input{2_related_work}
\input{3_pre}
\input{4_main_results}
\input{5.5_experiments}
\input{6_conclusion}

\newpage
\subsubsection*{Acknowledgements}
The authors would like to thank the anonymous reviewer of UAI 2025 for their highly insightful suggestions.

\ifdefined\isarxiv
%\section*{Acknowledgments}
\bibliographystyle{alpha}
\bibliography{ref}

\else

\bibliography{ref}
% \bibliographystyle{icml2025}
% \bibliographystyle{alpg mha}

\fi


\newpage
\onecolumn

\title{NRFlow: Towards Noise-Robust Generative Modeling via High-Order Flow
Matching (Supplementary Material)}
\maketitle
\appendix


\input{10_appendix_3rd}
\input{12_experiments}


%%%% Cut-line between first 10 pages and appendix







%%% some writing rules

%% Writing rule for creating tags.
%% Tags :
%% Theorem    \ref{thm:bla_bla}
%% Lemma      \ref{lem:bla_bla}
%% Claim      \ref{cla:bla_bla}
%% Corollary  \ref{cor:bla_bla}
%% Fact       \ref{fac:bla_bla}
%% Definition \ref{def:bla_bla}
%% Section    \ref{sec:bla_bla}
%% Subsection \ref{sub:bla_bla}
%% Equation   \ref{eq:bla_bla}



\end{document}



%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
