{"title":"改进斯坦因变分梯度下降的有限粒子收敛速率","authors":"Krishnakumar Balasubramanian, Sayan Banerjee, Promit Ghosal","doi":"arxiv-2409.08469","DOIUrl":null,"url":null,"abstract":"We provide finite-particle convergence rates for the Stein Variational\nGradient Descent (SVGD) algorithm in the Kernel Stein Discrepancy\n($\\mathsf{KSD}$) and Wasserstein-2 metrics. Our key insight is the observation\nthat the time derivative of the relative entropy between the joint density of\n$N$ particle locations and the $N$-fold product target measure, starting from a\nregular initial distribution, splits into a dominant `negative part'\nproportional to $N$ times the expected $\\mathsf{KSD}^2$ and a smaller `positive\npart'. This observation leads to $\\mathsf{KSD}$ rates of order $1/\\sqrt{N}$,\nproviding a near optimal double exponential improvement over the recent result\nby~\\cite{shi2024finite}. Under mild assumptions on the kernel and potential,\nthese bounds also grow linearly in the dimension $d$. By adding a bilinear\ncomponent to the kernel, the above approach is used to further obtain\nWasserstein-2 convergence. For the case of `bilinear + Mat\\'ern' kernels, we\nderive Wasserstein-2 rates that exhibit a curse-of-dimensionality similar to\nthe i.i.d. setting. We also obtain marginal convergence and long-time\npropagation of chaos results for the time-averaged particle laws.","PeriodicalId":501379,"journal":{"name":"arXiv - STAT - Statistics Theory","volume":"49 1","pages":""},"PeriodicalIF":0.0000,"publicationDate":"2024-09-13","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"Improved Finite-Particle Convergence Rates for Stein Variational Gradient Descent\",\"authors\":\"Krishnakumar Balasubramanian, Sayan Banerjee, Promit Ghosal\",\"doi\":\"arxiv-2409.08469\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"We provide finite-particle convergence rates for the Stein Variational\\nGradient Descent (SVGD) algorithm in the Kernel Stein Discrepancy\\n($\\\\mathsf{KSD}$) and Wasserstein-2 metrics. Our key insight is the observation\\nthat the time derivative of the relative entropy between the joint density of\\n$N$ particle locations and the $N$-fold product target measure, starting from a\\nregular initial distribution, splits into a dominant `negative part'\\nproportional to $N$ times the expected $\\\\mathsf{KSD}^2$ and a smaller `positive\\npart'. This observation leads to $\\\\mathsf{KSD}$ rates of order $1/\\\\sqrt{N}$,\\nproviding a near optimal double exponential improvement over the recent result\\nby~\\\\cite{shi2024finite}. Under mild assumptions on the kernel and potential,\\nthese bounds also grow linearly in the dimension $d$. By adding a bilinear\\ncomponent to the kernel, the above approach is used to further obtain\\nWasserstein-2 convergence. For the case of `bilinear + Mat\\\\'ern' kernels, we\\nderive Wasserstein-2 rates that exhibit a curse-of-dimensionality similar to\\nthe i.i.d. setting. We also obtain marginal convergence and long-time\\npropagation of chaos results for the time-averaged particle laws.\",\"PeriodicalId\":501379,\"journal\":{\"name\":\"arXiv - STAT - Statistics Theory\",\"volume\":\"49 1\",\"pages\":\"\"},\"PeriodicalIF\":0.0000,\"publicationDate\":\"2024-09-13\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"arXiv - STAT - Statistics Theory\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/arxiv-2409.08469\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"\",\"JCRName\":\"\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"arXiv - STAT - Statistics Theory","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/arxiv-2409.08469","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
Improved Finite-Particle Convergence Rates for Stein Variational Gradient Descent
We provide finite-particle convergence rates for the Stein Variational
Gradient Descent (SVGD) algorithm in the Kernel Stein Discrepancy
($\mathsf{KSD}$) and Wasserstein-2 metrics. Our key insight is the observation
that the time derivative of the relative entropy between the joint density of
$N$ particle locations and the $N$-fold product target measure, starting from a
regular initial distribution, splits into a dominant `negative part'
proportional to $N$ times the expected $\mathsf{KSD}^2$ and a smaller `positive
part'. This observation leads to $\mathsf{KSD}$ rates of order $1/\sqrt{N}$,
providing a near optimal double exponential improvement over the recent result
by~\cite{shi2024finite}. Under mild assumptions on the kernel and potential,
these bounds also grow linearly in the dimension $d$. By adding a bilinear
component to the kernel, the above approach is used to further obtain
Wasserstein-2 convergence. For the case of `bilinear + Mat\'ern' kernels, we
derive Wasserstein-2 rates that exhibit a curse-of-dimensionality similar to
the i.i.d. setting. We also obtain marginal convergence and long-time
propagation of chaos results for the time-averaged particle laws.