collaborators

14 papers

cs.LG2026

An AI4AI Framework for Visual Token Pruning

Zhen Liu, Wenli Huang, Wei Song +3

Visual-token pruning can substantially reduce the inference cost of multimodal large language models (MLLMs), yet existing methods largely rely on fixed, handcrafted heuristics and…

cs.CV2026

Overcoming Statistical Bias in Action-Controllable World Models

Yuhong Shi, Zhenhao Chu, Jie Wei +3

Action-conditioned world models aim to predict how visual environments evolve under an agent's actions. Yet future frames are often highly predictable from visual inertia and recur…

cs.AI2026

Improving Auto-Design of Neural PDE Solvers with a Domain-Specific Language

Shengxin Kong, Liwen Xu, Jingwen Fu

Neural PDE solver auto-design is fundamentally a search-space representation problem. In the space of unrestricted Python programs, valid solvers form an extremely sparse subset: m…

cs.LG2026

GraphIR: Architecture-Level Search States for LLM-Guided Neural Architecture Evolution

Zhen Liu, Wanqi Zhou, Shuanghao Bai +3

Large language models (LLMs) enable neural architecture search (NAS) directly over executable neural network programs. However, code-level flexibility does not provide the architec…

cs.LG2026

Overcoming the Weakest-Link Effect in LLM-Driven Program Optimization via Heterogeneous Edit Recombination

Jingwen Fu, Zhen Liu, Yuhan Liu +2

Large language models (LLMs) are increasingly used to solve complex problems by searching over program space, offering a general paradigm for scientific problems that can be natura…

cs.IR2026

ELVA: Exploring Ranking-Driven Universal Multimodal Retrieval

Yuhan Liu, Pei Fu, Hang Li +8

Leveraging Multimodal Large Language Models (MLLMs) via contrastive learning has become a mainstream paradigm for improving the performance of Universal Multimodal Retrieval (UMR).…