32 citations · 32 across the 2 of their papers we have counts for
4 papers · 1 filter
DMPlug: A Plug-in Method for Solving Inverse Problems with Diffusion Models
Hengkang Wang, Xu Zhang, Taihui Li +3
Pretrained diffusion models (DMs) have recently been popularly used in solving inverse problems (IPs). The existing methods mostly interleave iterative steps in the reverse diffusi…
De-randomized PAC-Bayes Margin Bounds: Applications to Non-convex and Non-smooth Predictors
Arindam Banerjee, Tiancong Chen, Yingxue Zhou
In spite of several notable efforts, explaining the generalization of deterministic non-smooth deep nets, e.g., ReLU-nets, has remained challenging. Existing approaches for determi…
Hessian based analysis of SGD for Deep Nets: Dynamics and Generalization
Xinyan Li, Qilong Gu, Yingxue Zhou +2
While stochastic gradient descent (SGD) and variants have been surprisingly successful for training deep nets, several aspects of the optimization dynamics and generalization are s…
Distributed Training with Heterogeneous Data: Bridging Median- and Mean-Based Algorithms
Xiangyi Chen, Tiancong Chen, Haoran Sun +2
Recently, there is a growing interest in the study of median-based algorithms for distributed non-convex optimization. Two prominent such algorithms include signSGD with majority v…