← Search

Jiahang Xu

4 accepted papers

2025

Mutual Reasoning Makes Smaller LLMs Stronger Problem-Solver

ICLR 2025poster

This paper introduces rStar, a self-play mutual reasoning approach that significantly improves reasoning capabilities of small language models (SLMs) without fine-tuning or superior models. rStar decouples reasoning into a self-play mutual generation-discrimination process. First, a target SLM augme…

2024

LongRoPE: Extending LLM Context Window Beyond 2 Million Tokens

ICML 2024poster

Large context window is a desirable feature in large language models (LLMs). However, due to high fine-tuning costs, scarcity of long texts, and catastrophic values introduced by new token positions, current extended context windows are limited to around 128k tokens. This paper introduces LongRoPE t…

2023

ElasticViT: Conflict-aware Supernet Training for Deploying Fast Vision Transformer on Diverse Mobile Devices

ICCV 2023poster

Neural Architecture Search (NAS) has shown promising performance in the automatic design of vision transformers (ViT) exceeding 1G FLOPs. However, designing lightweight and low-latency ViT models for diverse mobile devices remains a big challenge. In this work, we propose ElasticViT, a two-stage NAS…

Cited by 25PDFcodeScholar
2023

SpaceEvo: Hardware-Friendly Search Space Design for Efficient INT8 Inference

ICCV 2023poster

The combination of Neural Architecture Search (NAS) and quantization has proven successful in automatically designing low-FLOPs INT8 quantized neural networks (QNN). However, directly applying NAS to design accurate QNN models that achieve low latency on real-world devices leads to inferior performa…

Cited by 5PDFcodeScholar