← Search

Yifei Cheng

4 accepted papers

2026

SCNS: Continual Personalization of Diffusion Models via Submodular Concept Neuron Selection

ICML 2026poster

Custom diffusion models (CDMs) have demonstrated impressive success in visual personalization tasks by enabling the generation of user-specific concepts. However, existing CDMs typically assume that personalized concepts are static and rely on costly model merging or sequential updates that are pron…

Cited by 0SourceScholar
2025

Knocking on IP: Unveiling Websites through Cache-Aware Fingerprinting

ICASSP 2025accepted

As user privacy becomes increasingly critical in the digital landscape, traditional methods of website fingerprinting (WF) face significant challenges, particularly in caching scenarios. Existing WF studies are limited by the assumption of disabled caching. Recently, only a few have explored how to…

Cited by 0SourceScholar
2023

DropIT: Dropping Intermediate Tensors for Memory-Efficient DNN Training

ICLR 2023poster

A standard hardware bottleneck when training deep neural networks is GPU memory. The bulk of memory is occupied by caching intermediate tensors for gradient computation in the backward pass. We propose a novel method to reduce this footprint - Dropping Intermediate Tensors (DropIT). DropIT drops mi…

2021

STL-SGD: Speeding Up Local SGD with Stagewise Communication Period

AAAI 2021technical

Distributed parallel stochastic gradient descent algorithms are workhorses for large scale machine learning tasks. Among them, local stochastic gradient descent (Local SGD) has attracted significant attention due to its low communication complexity. Previous studies prove that the communication comp…

Cited by 8SourcePDFScholar