← Search

Kai Yi

14 accepted papers

2026

CARE: A Molecular-Guided Foundation Model with Adaptive Region Modeling for Whole Slide Image Analysis

CVPR 2026

Foundation models have achieved success in computational pathology, demonstrating generalization across histopathology tasks. However, existing models overlook the heterogeneous and non-uniform organization of regions of interest (ROIs) because they rely on natural image backbones not tailored for t

Cited by 0SourcecodeScholar
2026

WinQ: Accelerating Quantization-Aware Training of Large Language Models around Saddle Points

ICML 2026poster

Quantization-aware training is widely used for language model quantization in sub-4-bit precision, by training full-precision weights with gradients computed on the quantized model. The main bottleneck for this training approach is its slow convergence and plateauing of test performance, which gets …

Cited by 0SourceScholar
2025

How Particle System Theory Enhances Hypergraph Message Passing

NeurIPS 2025poster

Hypergraphs effectively model higher-order relationships in natural phenomena, capturing complex interactions beyond pairwise connections. We introduce a novel hypergraph message passing framework inspired by interacting particle systems, where hyperedges act as fields inducing shared node dynamics.…

Cited by 0SourcecodeScholar
2025

MANTA: A Large-Scale Multi-View and Visual-Text Anomaly Detection Dataset for Tiny Objects

CVPR 2025poster

We present MANTA, a visual-text anomaly detection dataset for tiny objects. The visual component comprises over 137.3K images across 38 object categories spanning five typical domains, of which 8.6K images are labeled as anomalous with pixel-level annotations. Each image is captured from five distin…

Cited by 1SourcePDFScholar
2024

FedP3: Federated Personalized and Privacy-friendly Network Pruning under Model Heterogeneity

ICLR 2024poster

The interest in federated learning has surged in recent research due to its unique ability to train a global model using privacy-secured information held locally on each client. This paper pays particular attention to the issue of client-side model heterogeneity, a pervasive challenge in the practic…

Cited by 13SourcePDFScholar
2024

PV-Tuning: Beyond Straight-Through Estimation for Extreme LLM Compression

NeurIPS 2024oral

There has been significant interest in "extreme" compression of large language models (LLMs), i.e. to 1-2 bits per parameter, which allows such models to be executed efficiently on resource-constrained devices. Existing work focused on improved one-shot quantization techniques and weight represent…

2023

ACMP: Allen-Cahn Message Passing with Attractive and Repulsive Forces for Graph Neural Networks

ICLR 2023top-25%

Neural message passing is a basic feature extraction unit for graph-structured data considering neighboring node features in network propagation from one layer to the next. We model such process by an interacting particle system with attractive and repulsive forces and the Allen-Cahn force arising i…

2023

Continual Zero-Shot Learning through Semantically Guided Generative Random Walks

ICCV 2023poster

Learning novel concepts, remembering previous knowledge, and adapting it to future tasks occur simultaneously throughout a human's lifetime. To model such comprehensive abilities, continual zero-shot learning (CZSL) has recently been introduced. However, most existing methods overused the unseen sem…

Cited by 6PDFcodeScholar
2023

Graph Denoising Diffusion for Inverse Protein Folding

NeurIPS 2023poster

Inverse protein folding is challenging due to its inherent one-to-many mapping characteristic, where numerous possible amino acid sequences can fold into a single, identical protein backbone. This task involves not only identifying viable sequences but also representing the sheer diversity of potent…

2022

EF-BV: A Unified Theory of Error Feedback and Variance Reduction Mechanisms for Biased and Unbiased Compression in Distributed Optimization

NeurIPS 2022accept

In distributed or federated optimization and learning, communication between the different computing units is often the bottleneck and gradient compression is widely used to reduce the number of bits sent within each communication round of iterative methods. There are two classes of compression oper…

2022

Exploring Hierarchical Graph Representation for Large-Scale Zero-Shot Image Classification

ECCV 2022poster

"The main question we address in this paper is how to scale up visual recognition of unseen classes, also known as zero-shot learning, to tens of thousands of categories as in the ImageNet-21K benchmark. At this scale, especially with many fine-grained categories included in ImageNet-21K, it is crit…

2022

Variance Reduced ProxSkip: Algorithm, Theory and Application to Federated Learning

NeurIPS 2022accept

We study distributed optimization methods based on the {\em local training (LT)} paradigm, i.e., methods which achieve communication efficiency by performing richer local gradient-based training on the clients before (expensive) parameter averaging is allowed to take place. While these methods were…

2022

VisualGPT: Data-Efficient Adaptation of Pretrained Language Models for Image Captioning

CVPR 2022poster

The limited availability of annotated data often hinders real-world applications of machine learning. To efficiently learn from small quantities of multimodal data, we leverage the linguistic knowledge from a large pre-trained language model (PLM) and quickly adapt it to new domains of image caption…

Cited by 277PDFcodeScholar