← Search

Ke Cheng

15 accepted papers

2026

Differentially Private Subspace Fine-Tuning for Large Language Models

AAAI 2026technical

Fine-tuning large language models on downstream tasks is crucial for realizing their cross-domain potential but often relies on sensitive data, raising privacy concerns. Differential privacy (DP) offers rigorous privacy guarantees and has been widely adopted in fine-tuning; however, naively injecti

Cited by 0SourcePDFScholar
2026

Federated Manifold Learning (FML): Tackling Domain Heterogeneity with Structural Knowledge Transfer

ICML 2026poster

Federated Learning (FL) faces significant challenges due to domain heterogeneity, where data from different clients exhibit substantial statistical shifts that hinder the generalization of the global model. Although existing methods attempt to mitigate this by exchanging class prototypes, they fall …

Cited by 0SourceScholar
2026

PENTESTLLMAGENT: A Task Dependency Graph Planning-Based Multi-Agent Framework for Automated Penetration Testing

IJCAI 2026

Fully autonomous IP-to-Root penetration testing remains challenging for LLM agents. We conduct an exploratory study on 10 LLMs and introduce AutoPentest-Bench, an end-to-end benchmark with 13 VulnHub targets and 93 sub-tasks. From 130 interaction logs, we identify three challenges: Rigid Strategy, C

Cited by 0Scholar
2026

PRISM: Sequence Modeling as Parallel Residual Iteration

ICML 2026poster

Generative sequence modeling faces a fundamental tension between the expressivity of Transformers and the efficiency of linear sequence models. Existing efficient architectures are theoretically bounded by shallow, single-step linear updates, while powerful iterative methods like Test-Time Training …

Cited by 0SourceScholar
2026

Proteus: Lookup-Free Trellis-Coded Quantization by Lattice-Breaking Compute Codes for 2-Bit LLMs

ICML 2026poster

Autoregressive decoding of large language models is frequently memory-traffic bound, so ultra-low-bit weight-only PTQ helps only if dequantization avoids irregular codebook or LUT access in the inner loop. Under the GPU-friendly bitshift trellis, existing 2-bit trellis-coded quantization (TCQ) pipel…

Cited by 0SourceScholar
2026

UNO! UNified Offline Training Paradigm for Learning Path Recommendation

AAAI 2026technical

With the wide adoption of online education platforms, adaptive learning systems have become increasingly important. Learning Path Recommendation (LPR) aims to dynamically adjust learning content to optimize learning efficiency based on individual student needs. However, current LPR methods suffer fr

Cited by 0SourcePDFScholar
2025

CAKE: Cascading and Adaptive KV Cache Eviction with Layer Preferences

ICLR 2025poster

Large language models (LLMs) excel at processing long sequences, boosting demand for key-value (KV) caching. While recent efforts to evict KV cache have alleviated the inference burden, they often fail to allocate resources rationally across layers with different attention patterns. In this paper, w…

2024

DrivingDojo Dataset: Advancing Interactive and Knowledge-Enriched Driving World Model

NeurIPS 2024poster

Driving world models have gained increasing attention due to their ability to model complex physical dynamics. However, their superb modeling capability is yet to be fully unleashed due to the limited video diversity in current driving datasets. We introduce DrivingDojo, the first dataset tailor-mad…

Cited by 7SourcePDFScholar
2024

Securely and Efficiently Outsourcing Neural Network Inference via Parallel MSB Extraction

ICASSP 2024accepted

Outsourcing neural network (NN) inference services to the cloud gives rise to considerable privacy concerns about the model provider’s proprietary model and the user’s private data. Current cryptography-based secure NN inference schemes are not suited for high-latency networks due to their numerous…

Cited by 0SourceScholar
2022

MENet: A Memory-Based Network with Dual-Branch for Efficient Event Stream Processing

ECCV 2022poster

"Event cameras are bio-inspired sensors that asynchronously capture per-pixel brightness change and trigger a stream of events instead of frame-based images. Each event stream is generally split into multiple sliding windows for subsequent processing. However, most existing event-based methods ignor…

Cited by 1SourcePDFScholar
2022

PKD: General Distillation Framework for Object Detectors via Pearson Correlation Coefficient

NeurIPS 2022accept

Knowledge distillation(KD) is a widely-used technique to train compact models in object detection. However, there is still a lack of study on how to distill between heterogeneous detectors. In this paper, we empirically find that better FPN features from a heterogeneous teacher detector can help the…

2022

Towards Fully Sparse Training: Information Restoration with Spatial Similarity

AAAI 2022technical

The 2:4 structured sparsity pattern released by NVIDIA Ampere architecture, requiring four consecutive values containing at least two zeros, enables doubling math throughput for matrix multiplications. Recent works mainly focus on inference speedup via 2:4 sparsity while training acceleration has be…

Cited by 2SourcePDFScholar
2020

Decoupling GCN with DropGraph Module for Skeleton-Based Action Recognition

ECCV 2020poster

In skeleton-based action recognition, graph convolutional networks (GCNs) have achieved remarkable success. Nevertheless, how to efficiently model the spatial-temporal skeleton graph without introducing extra computation burden is a challenging problem for industrial deployment. In this paper, we re…

2020

ProxyBNN: Learning Binarized Neural Networks via Proxy Matrices

ECCV 2020poster

Training Binarized Neural Networks (BNNs) is challenging due to the discreteness. In order to efficiently optimize BNNs through backward propagations, real-valued auxiliary variables are commonly used to accumulate gradient updates. Those auxiliary variables are then directly quantized to binary wei…

Cited by 36SourcePDFScholar
2020

Skeleton-Based Action Recognition With Shift Graph Convolutional Network

CVPR 2020oral

Action recognition with skeleton data is attracting more attention in computer vision. Recently, graph convolutional networks (GCNs), which model the human body skeletons as spatiotemporal graphs, have obtained remarkable performance. However, the computational complexity of GCN-based methods are pr…

Cited by 1008PDFScholar