← Search

Shi Pu

6 accepted papers

2026

BAS: Bridging Adam and SignSGD for Memory-Efficient LLM Training

ICML 2026poster

We propose **Block Adaptive Signum (BAS)**, which bridges Adam and SignSGD via block-wise scaling of sign updates. By discarding element-wise second moments, BAS reduces memory overhead relative to AdamW without sacrificing performance. Crucially, BAS mimics Adam’s dynamics closely enough to directl…

Cited by 0SourceScholar
2024

B-ary Tree Push-Pull Method is Provably Efficient for Distributed Learning on Heterogeneous Data

NeurIPS 2024poster

This paper considers the distributed learning problem where a group of agents cooperatively minimizes the summation of their local cost functions based on peer-to-peer communication. Particularly, we propose a highly efficient algorithm, termed ``B-ary Tree Push-Pull'' (BTPP), that employs two B-ary…

2023

One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale

ICML 2023poster

This paper proposes a unified diffusion framework (dubbed UniDiffuser) to fit all distributions relevant to a set of multi-modal data in one model. Our key insight is -- learning diffusion models for marginal, conditional, and joint distributions can be unified as predicting the noise in the perturb…

2022

Alignment-Uniformity Aware Representation Learning for Zero-Shot Video Classification

CVPR 2022poster

Most methods tackle zero-shot video classification by aligning visual-semantic representations within seen classes, which limits generalization to unseen classes. To enhance model generalizability, this paper presents an end-to-end framework that preserves alignment and uniformity properties for rep…

Cited by 27PDFcodeScholar
2018

Deep Attentive Tracking via Reciprocative Learning

NeurIPS 2018poster

Visual attention, derived from cognitive neuroscience, facilitates human perception on the most pertinent subset of the sensory data. Recently, significant efforts have been made to exploit attention schemes to advance computer vision systems. For visual tracking, it is often challenging to track ta…

Cited by 225SourcePDFScholar