The Langevin Dynamics (LD), which aims to sample from a probability distribution using its score function, has been widely used for analyzing and developing score-based generative modeling algorithms. While the convergence behavior of LD in sampling from a uni-modal distribution has been extensively studied in the literature, the analysis of LD under a mixture distribution with distinct modes remains underexplored in the literature. In this work, we analyze LD in sampling from a mixture distribution and theoretically study its convergence properties. Our theoretical results indicate that for general mixture distributions of sub-Gaussian components, LD could fail in finding all the components within a sub-exponential number of steps in the data dimension. Following our result on the complexity of LD in sampling from high-dimensional variables, we propose Chained Langevin Dynamics (Chained-LD), which divides the data vector into patches of smaller sizes and generates every patch sequentially conditioned on the previous patches. Our theoretical analysis of Chained-LD indicates its faster convergence speed to the components of a mixture distribution. We present the results of several numerical experiments on synthetic and real image datasets, validating our theoretical results on the iteration complexities of sample generation from mixture distributions using the vanilla and chained LD algorithms.
Department of Mathematics, University of California, Los Angeles, CA 90095（加州大学洛杉矶分校数学系）；Department of Mathematics, University of South Carolina, Columbia, SC 29208（南卡罗来纳大学数学系）
Plug-and-Play Posterior Sampling under Mismatched Measurement and Prior Models
Department of Electrical & Systems Engineering, Washington University in St. Louis, USA（华盛顿大学圣路易斯分校电气与系统工程系）；Computer Science Department, ENS, PSL University, 75005, France（巴黎高等师范学院计算机科学系）；MAP5, CNRS, Université Paris Cité, 75006, France（巴黎城市大学MAP5研究所）；Department of Computer Science & Engineering, Washington University in St. Louis, USA（华盛顿大学圣路易斯分校计算机科学与工程系）
The Hong Kong University of Science and Technology（香港科技大学）；Salesforce AI Research（Salesforce AI研究院）；University of California, San Diego（加利福尼亚大学圣迭戈分校）；University of Illinois Urbana-Champaign（伊利诺伊大学厄巴纳-香槟分校）
