← Search

Puyu Wang

9 accepted papers

2026

Landmark-Guided Policy Optimization for Multi-Objective Language Model Selection

ICML 2026poster

Selecting a pretrained large language model (LLM) to fine-tune for a task-specific dataset can be time-consuming and costly. With several candidate models available to choose from, varying in size, architecture, and pretraining data, finding the best model for a specific task often involves extensiv…

Cited by 0SourceScholar
2026

Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov–Arnold Networks

ICML 2026poster

Kolmogorov--Arnold Networks (KANs) have recently emerged as a structured alternative to standard MLPs, yet a principled theory for their training dynamics, generalization, and privacy properties remains limited. In this paper, we analyze gradient descent (GD) for training two-layer KANs and derive g…

Cited by 0SourceScholar
2026

Reimagining Anomalies: What If Anomalies Were Normal?

AAAI 2026technical

Deep learning-based methods have achieved a breakthrough in image anomaly detection, but their complexity introduces a considerable challenge to understanding why an instance is predicted to be anomalous. We introduce a novel explanation method that generates multiple alternative modifications for e

Cited by 0SourcePDFScholar
2026

TORA: Train Once, Realign Anytime for Offline Multi-Objective Reinforcement Learning

AAAI 2026technical

Intelligent agents in real-world applications must adapt their behavior to changing contexts and user preferences. For example, planning a road trip requires considering both travel time and cost. Multi-objective reinforcement learning (MORL) provides a principled approach to navigate such trade-of

Cited by 0SourcePDFScholar
2026

Towards Understanding Generalization in DP-GD: A Case Study in Training Two-Layer CNNs

AAAI 2026technical

Modern deep learning techniques focus on extracting intricate information from data to achieve accurate predictions. However, the training datasets may be crowdsourced and include sensitive information, such as personal contact details, financial data, and medical records. As a result, there is a gr

Cited by 0SourcePDFScholar
2024

Non-vacuous Generalization Bounds for Adversarial Risk in Stochastic Neural Networks

AISTATS 2024poster

Adversarial examples are manipulated samples used to deceive machine learning models, posing a serious threat in safety-critical applications. Existing safety certificates for machine learning models are limited to individual input examples, failing to capture generalization to unseen data. To addre…

2022

Stability and Generalization for Markov Chain Stochastic Gradient Methods

NeurIPS 2022accept

Recently there is a large amount of work devoted to the study of Markov chain stochastic gradient methods (MC-SGMs) which mainly focus on their convergence analysis for solving minimization problems. In this paper, we provide a comprehensive generalization analysis of MC-SGMs for both minimization…

Cited by 21SourcePDFScholar
2021

Simple Stochastic and Online Gradient Descent Algorithms for Pairwise Learning

NeurIPS 2021poster

Pairwise learning refers to learning tasks where the loss function depends on a pair of instances. It instantiates many important machine learning tasks such as bipartite ranking and metric learning. A popular approach to handle streaming data in pairwise learning is an online gradient descent (OG…