MARS: Unleashing the Power of Variance Reduction for Training Large Models Nov 15, 2024 Stochastic Optimization
Code Code Available Benchopt: Reproducible, efficient and collaborative optimization benchmarks Jun 27, 2022 Benchmarking image-classification
Code Code Available Conformal Symplectic Optimization for Stable Reinforcement Learning Dec 3, 2024 Atari Games Deep Reinforcement Learning
Code Code Available Smoothing Methods for Automatic Differentiation Across Conditional Branches Oct 5, 2023 Stochastic Optimization
Code Code Available Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training May 23, 2023 Language Modeling Language Modelling
Code Code Available Riemannian Adaptive Optimization Methods Oct 1, 2018 Riemannian optimization Stochastic Optimization
Code Code Available Accurate, Large Minibatch SGD: Training ImageNet in 1 Hour Jun 8, 2017 Stochastic Optimization
Code Code Available Adaptive Semantic Token Communication for Transformer-based Edge Inference May 23, 2025 object-detection Object Detection
Code Code Available JaxSGMC: Modular stochastic gradient MCMC in JAX May 16, 2025 Deep Learning Stochastic Optimization
Code Code Available A Novel Unified Parametric Assumption for Nonconvex Optimization Feb 17, 2025 Stochastic Optimization
Code Code Available Time-Causal VAE: Robust Financial Time Series Generator Nov 5, 2024 Decoder Stochastic Optimization
Code Code Available Training-free Diffusion Model Alignment with Sampling Demons Oct 8, 2024 Denoising Image Generation
Code Code Available Randomized Physics-Informed Neural Networks for Bayesian Data Assimilation Jul 5, 2024 Stochastic Optimization Uncertainty Quantification
Code Code Available SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning Apr 28, 2024 Stochastic Optimization
Code Code Available The Acquisition of Physical Knowledge in Generative Neural Networks Oct 30, 2023 Stochastic Optimization
Code Code Available Why Do We Need Weight Decay in Modern Deep Learning? Oct 6, 2023 Deep Learning image-classification
Code Code Available Monte Carlo Policy Gradient Method for Binary Optimization Jul 3, 2023 Combinatorial Optimization Diversity
Code Code Available BinaryViT: Pushing Binary Vision Transformers Towards Convolutional Models Jun 29, 2023 Binarization Image Classification
Code Code Available Neural Inventory Control in Networks via Hindsight Differentiable Policy Optimization Jun 20, 2023 Deep Reinforcement Learning Management
Code Code Available MoMo: Momentum Models for Adaptive Learning Rates May 12, 2023 Recommendation Systems Stochastic Optimization
Code Code Available A Variational Perspective on Solving Inverse Problems with Diffusion Models May 7, 2023 Denoising Image Restoration
Code Code Available Decouple Graph Neural Networks: Train Multiple Simple GNNs Simultaneously Instead of One Apr 20, 2023 Stochastic Optimization
Code Code Available Combinatorial Optimization enriched Machine Learning to solve the Dynamic Vehicle Routing Problem with Time Windows Apr 3, 2023 Combinatorial Optimization Stochastic Optimization
Code Code Available Variational Linearized Laplace Approximation for Bayesian Deep Learning Feb 24, 2023 Deep Learning Stochastic Optimization
Code Code Available Optimal Planning of Hybrid Energy Storage Systems using Curtailed Renewable Energy through Deep Reinforcement Learning Dec 12, 2022 Decision Making Deep Reinforcement Learning
Code Code Available End-to-End Stochastic Optimization with Energy-Based Model Nov 25, 2022 Scheduling Stochastic Optimization
Code Code Available Stochastic Gradient Descent Captures How Children Learn About Physics Sep 25, 2022 Stochastic Optimization
Code Code Available Sequential Manipulation Planning on Scene Graph Jul 10, 2022 Object Rearrangement Stochastic Optimization
Code Code Available Communication-Efficient Adaptive Federated Learning May 5, 2022 Federated Learning Quantization
Code Code Available Exploiting Explainable Metrics for Augmented SGD Mar 31, 2022 Stochastic Optimization
Code Code Available A Framework for Improving the Reliability of Black-box Variational Inference Mar 29, 2022 Bayesian Inference Stochastic Optimization
Code Code Available Provable Stochastic Optimization for Global Contrastive Learning: Small Batch Does Not Harm Performance Feb 24, 2022 Contrastive Learning Self-Supervised Learning
Code Code Available Adapting to Mixing Time in Stochastic Optimization with Markovian Data Feb 9, 2022 Stochastic Optimization
Code Code Available Reinforcement Learning with Dynamic Convex Risk Measures Dec 26, 2021 reinforcement-learning Reinforcement Learning
Code Code Available BCD Nets: Scalable Variational Approaches for Bayesian Causal Discovery Dec 6, 2021 Causal Discovery Stochastic Optimization
Code Code Available Efficient approximation of Jacobian matrices involving a non-uniform fast Fourier transform (NUFFT) Nov 4, 2021 compressed sensing Image Reconstruction
Code Code Available slimTrain -- A Stochastic Approximation Method for Training Separable Deep Neural Networks Sep 28, 2021 Sensitivity Stochastic Optimization
Code Code Available ATD: Augmenting CP Tensor Decomposition by Self Supervision Jun 15, 2021 Data Augmentation Dimensionality Reduction
Code Code Available Scaling Up Graph Neural Networks Via Graph Coarsening Jun 9, 2021 Stochastic Optimization
Code Code Available Differentiable Quality Diversity Jun 7, 2021 Diversity Stochastic Optimization
Code Code Available Efficient Stochastic Optimal Control through Approximate Bayesian Input Inference May 17, 2021 Stochastic Optimization
Code Code Available Stochastic Optimization of Areas Under Precision-Recall Curves with Provable Convergence Apr 18, 2021 Stochastic Optimization
Code Code Available A Stochastic Optimization Framework for Fair Risk Minimization Feb 24, 2021 Binary Classification Fairness
Code Code Available Sinkhorn Label Allocation: Semi-Supervised Classification via Annealed Self-Training Feb 17, 2021 Classification General Classification
Code Code Available Parameter-free Stochastic Optimization of Variationally Coherent Functions Jan 30, 2021 Stochastic Optimization
Code Code Available Adaptivity without Compromise: A Momentumized, Adaptive, Dual Averaged Gradient Method for Stochastic Optimization Jan 26, 2021 Stochastic Optimization
Code Code Available LIRA: Learnable, Imperceptible and Robust Backdoor Attacks Jan 1, 2021 Backdoor Attack backdoor defense
Code Code Available Stochastic Gradient Variance Reduction by Solving a Filtering Problem Dec 22, 2020 Stochastic Optimization
Code Code Available Learning from History for Byzantine Robust Optimization Dec 18, 2020 Federated Learning Stochastic Optimization
Code Code Available Quality-Diversity Optimization: a novel branch of stochastic optimization Dec 8, 2020 Diversity Stochastic Optimization
Code Code Available 