← Search

Mingjia Shi

9 accepted papers

2026

IAPO: Information-Aware Policy Optimization for Token-Efficient Reasoning

ICML 2026poster

Large language models increasingly rely on long chains of thought to improve accuracy, yet such gains come with substantial inference-time costs. We revisit token-efficient post-training and argue that existing sequence-level reward-shaping methods offer limited control over how reasoning effort is …

Cited by 0SourceScholar
2025

Drag-and-Drop LLMs: Zero-Shot Prompt-to-Weights

NeurIPS 2025poster

Modern Parameter-Efficient Fine-Tuning (PEFT) methods such as low-rank adaptation (LoRA) reduce the cost of customizing large language models (LLMs), yet still require a separate optimization run for every downstream dataset. We introduce \textbf{Drag-and-Drop LLMs (\textit{DnD})}, a prompt-conditio…

Cited by 0SourcecodeScholar
2025

Ferret: An Efficient Online Continual Learning Framework under Varying Memory Constraints

CVPR 2025poster

In the realm of high-frequency data streams, achieving real-time learning within varying memory constraints is paramount. This paper presents Ferret, a comprehensive framework designed to enhance online accuracy of Online Continual Learning (OCL) algorithms while dynamically adapting to varying memo…

Cited by 0SourcePDFScholar
2025

GSQ-Tuning: Group-Shared Exponents Integer in Fully Quantized Training for LLMs On-Device Fine-tuning

ACL 2025finding

Large Language Models (LLMs) fine-tuning technologies have achieved remarkable results. However, traditional LLM fine-tuning approaches face significant challenges: they require large Floating Point(FP) computation, raising privacy concerns when handling sensitive data, and are impractical for resou…

2025

Pruning-Robust Mamba with Asymmetric Multi-Scale Scanning Paths

NeurIPS 2025poster

Mamba has proven efficient for long-sequence modeling in vision tasks. However, when token reduction techniques are applied to improve efficiency, Mamba-based models exhibit drastic performance degradation compared to Vision Transformers (ViTs). This decline is potentially attributed to Mamba's cha…

Cited by 0SourceScholar
2025

REPA Works Until It Doesn’t: Early-Stopped, Holistic Alignment Supercharges Diffusion Training

NeurIPS 2025poster

Diffusion Transformers (DiTs) deliver state-of-the-art image quality, yet their training remains notoriously slow. A recent remedy---representation alignment (REPA) that matches DiT hidden features to those of a non-generative teacher (e.g., DINO)---dramatically accelerates the early epochs but plat…

Cited by 0SourcecodeScholar
2025

Tackling Feature-Classifier Mismatch in Federated Learning via Prompt-Driven Feature Transformation

NeurIPS 2025poster

Federated Learning (FL) faces challenges due to data heterogeneity, which limits the global model’s performance across diverse client distributions. Personalized Federated Learning (PFL) addresses this by enabling each client to process an individual model adapted to its local distribution. Many exi…

Cited by 0SourceScholar
2023

Communication-efficient Federated Learning with Single-Step Synthetic Features Compressor for Faster Convergence

ICCV 2023poster

Reducing communication overhead in federated learning (FL) is challenging but crucial for large-scale distributed privacy-preserving machine learning. While methods utilizing sparsification or other techniques can largely reduce the communication overhead, the convergence rate is also greatly compro…

Cited by 12PDFcodeScholar
2023

PRIOR: Personalized Prior for Reactivating the Information Overlooked in Federated Learning.

NeurIPS 2023poster

Classical federated learning (FL) enables training machine learning models without sharing data for privacy preservation, but heterogeneous data characteristic degrades the performance of the localized model. Personalized FL (PFL) addresses this by synthesizing personalized models from a global mode…