← Search

Wei Shi

32 accepted papers

2026

Caracal: Causal Architecture via Spectral Mixing

ICML 2026poster

The scalability of Large Language Models to long sequences is hindered by the quadratic cost of self-attention and the limitations of positional encodings. To address these, we introduce **Caracal**, a novel architecture that replaces self-attention with a parameter-efficient, $\mathcal{O}(L \log L)…

Cited by 0SourceScholar
2026

HuiduRep: A Robust Self-Supervised Framework for Learning Neural Representations from Extracellular Recordings

AAAI 2026technical

Extracellular recordings are transient voltage fluctuations in the vicinity of neurons, serving as a fundamental modality in neuroscience for decoding brain activity at single-neuron resolution. Spike sorting, the process of attributing each detected spike to its corresponding neuron, is a pivotal s

Cited by 0SourcePDFScholar
2026

Interpretable Reward Model via Sparse Autoencoder

AAAI 2026technical

Large language models (LLMs) have been widely deployed across numerous fields. Reinforcement Learning from Human Feedback (RLHF) leverages reward models (RMs) as proxies for human preferences to align LLM behaviors with human values, making the accuracy, reliability, and interpretability of RMs crit

Cited by 0SourcePDFScholar
2026

REVIS: Sparse Latent Steering to Mitigate Object Hallucination in Large Vision-Language Models

ICML 2026poster

Despite the advanced capabilities of Large Vision-Language Models (LVLMs), they frequently suffer from object hallucination. One reason is that visual features and pretrained textual representations often become intertwined in the deeper network layers. To address this, we propose REVIS, a training-…

Cited by 0SourceScholar
2025

A Survey of Optimization Modeling Meets LLMs: Progress and Future Directions

IJCAI 2025

By virtue of its great utility in solving real-world problems, optimization modeling has been widely employed for optimal decision-making across various sectors, but it requires substantial expertise from operations research professionals. With the advent of large language models (LLMs), new opportu

Cited by 0SourcePDFScholar
2025

Advancing Sequential Numerical Prediction in Autoregressive Models

ACL 2025short

Autoregressive models have become the de facto choice for sequence generation tasks, but standard approaches treat digits as independent tokens and apply cross-entropy loss, overlooking the coherent structure of numerical sequences. This paper introduces Numerical Token Integrity Loss(NTIL) to addre…

2025

BPP-Search: Enhancing Tree of Thought Reasoning for Mathematical Modeling Problem Solving

ACL 2025long

LLMs exhibit advanced reasoning capabilities, offering the potential to transform natural language questions into mathematical models. However, existing open-source datasets in operations research domain lack detailed annotations of the modeling process, such as variable definitions, focusing solely…

2025

Dolphin: Document Image Parsing via Heterogeneous Anchor Prompting

ACL 2025finding

Document image parsing is challenging due to its complexly intertwined elements such as text paragraphs, figures, formulas, and tables. Current approaches either assemble specialized expert models or directly generate page-level content autoregressively, facing integration overhead, efficiency bottl…

2025

MASTER: A Multi-Agent System with LLM Specialized MCTS

NAACL 2025long

Large Language Models (LLM) are increasingly being explored for problem-solving tasks. However, their strategic planning capability is often viewed with skepticism. Recent studies have incorporated the Monte Carlo Tree Search (MCTS) algorithm to augment the planning capacity of LLM. Despite its pote…

Cited by 0SourcePDFScholar
2025

OptiBench Meets ReSocratic: Measure and Improve LLMs for Optimization Modeling

ICLR 2025poster

Large language models (LLMs) have exhibited their problem-solving abilities in mathematical reasoning. Solving realistic optimization (OPT) problems in application scenarios requires advanced and applied mathematics ability. However, current OPT benchmarks that merely solve linear programming are fa…

2025

ParGo: Bridging Vision-Language with Partial and Global Views

AAAI 2025technical

This work presents ParGo, a novel Partial-Global projector designed to connect the vision and language modalities for Multimodal Large Language Models (MLLMs). Unlike previous works that rely on global attention-based projectors, our ParGo bridges the representation gap between the separately pre-tr…

2025

Route Sparse Autoencoder to Interpret Large Language Models

EMNLP 2025

Mechanistic interpretability of large language models (LLMs) aims to uncover the internal processes of information propagation and reasoning. Sparse autoencoders (SAEs) have demonstrated promise in this domain by extracting interpretable and monosemantic features. However, prior works primarily focu

2024

Crafting Personalized Agents through Retrieval-Augmented Generation on Editable Memory Graphs

EMNLP 2024main

In the age of mobile internet, user data, often referred to as memories, is continuously generated on personal devices. Effectively managing and utilizing this data to deliver services to users is a compelling research topic. In this paper, we introduce a novel task of crafting personalized agents p…

2024

M-RAG: Reinforcing Large Language Model Performance through Retrieval-Augmented Generation with Multiple Partitions

ACL 2024long

Retrieval-Augmented Generation (RAG) enhances Large Language Models (LLMs) by retrieving relevant memories from an external database. However, existing RAG methods typically organize all memories in a whole database, potentially limiting focus on crucial memories and introducing noise. In this paper…

Cited by 20SourcePDFScholar
2024

Meta-Learning via PAC-Bayesian with Data-Dependent Prior: Generalization Bounds from Local Entropy

IJCAI 2024poster

Meta-learning accelerates the learning process on unseen learning tasks by acquiring prior knowledge through previous related tasks. The PAC-Bayesian theory provides a theoretical framework to analyze the generalization of meta-learning to unseen tasks. However, previous works still encounter two no…

Cited by 0SourcePDFScholar
2024

SEGMENT+: Long Text Processing with Short-Context Language Models

EMNLP 2024main

There is a growing interest in expanding the input capacity of language models (LMs) across various domains. However, simply increasing the context window does not guarantee robust performance across diverse long-input processing tasks, such as understanding extensive documents and extracting detail…

Cited by 1SourcePDFScholar
2023

Boosting Person Re-Identification with Viewpoint Contrastive Learning and Adversarial Training

ICASSP 2023accepted

Person re-identification (ReID) aims at retrieving a person of interest across multiple cameras. Despite significant progress in person ReID, viewpoint variation remains an obstacle to extracting discriminative features for retrieval. To address this problem, we propose a Viewpoint-Robust Network (V…

Cited by 0SourceScholar
2023

Investigating Transformer-Guided Chaining for Interpretable Natural Logic Reasoning

ACL 2023findings

Natural logic reasoning has received increasing attention lately, with several datasets and neural models proposed, though with limited success. More recently, a new class of works have emerged adopting a Neuro-Symbolic approach, called transformer guided chaining, whereby the idea is to iteratively…

Cited by 1SourcePDFScholar
2023

Learning Concordant Attention via Target-aware Alignment for Visible-Infrared Person Re-identification

ICCV 2023poster

Owing to the large distribution gap between the heterogeneous data in Visible-Infrared Person Re-identification (VI Re-ID), we point out that existing paradigms often suffer from the inter-modal semantic misalignment issue and thus fail to align and compare local details properly. In this paper, we…

Cited by 37PDFScholar
2023

Say What You Mean! Large Language Models Speak Too Positively about Negative Commonsense Knowledge

ACL 2023long

Large language models (LLMs) have been widely studied for their ability to store and utilize positive knowledge. However, negative knowledge, such as “lions don’t live in the ocean”, is also ubiquitous in the world but rarely mentioned explicitly in text. What do LLMs know about negative knowledge?T…

2022

E-KAR: A Benchmark for Rationalizing Natural Language Analogical Reasoning

ACL 2022findings

The ability to recognize analogies is fundamental to human cognition. Existing benchmarks to test word analogy do not reveal the underneath process of analogical reasoning of neural models. Holding the belief that models capable of reasoning should be right for the right reasons, we propose a first-…

Cited by 35SourcePDFScholar
2022

Pose-Guided Feature Disentangling for Occluded Person Re-identification Based on Transformer

AAAI 2022technical

Occluded person re-identification is a challenging task as human body parts could be occluded by some obstacles (e.g. trees, cars, and pedestrians) in certain scenes. Some existing pose-guided methods solve this problem by aligning body parts according to graph matching, but these graph-based method…

2021

Adversarial Feature Disentanglement for Long-Term Person Re-identification

IJCAI 2021poster

Most existing person re-identification methods are effective in short-term scenarios because of their appearance dependencies. However, these methods may fail in long-term scenarios where people might change their clothes. To this end, we propose an adversarial feature disentanglement network (AFD-N…

Cited by 62SourcePDFScholar
2021

Modality-aware Style Adaptation for RGB-Infrared Person Re-Identification

IJCAI 2021poster

RGB-infrared (IR) person re-identification is a challenging task due to the large modality gap between RGB and IR images. Many existing methods bridge the modality gap by style conversion, requiring high-similarity images exchanged by complex CNN structures, like GAN. In this paper, we propose a hig…

Cited by 22SourcePDFScholar
2020

A Fast and Accurate Super-Resolution Network Using Progressive Residual Learning

ICASSP 2020accepted

Single-image super-resolution (SISR) task has witnessed great strides in the past few years with the development of deep learning. However, most existing studies concentrate on exploiting much deeper super-resolution networks, which are not friendly to the constrained computation resources. In this…

Cited by 0SourceScholar
2018

A Discriminatively Learned Feature Embedding Based on Multi-Loss Fusion For Person Search

ICASSP 2018accepted

Person search is a challenging task that requires to address pedestrian detection and person re- identification simultaneously. Though significant progress has been made in detection and re-identification respectively, the similar appearances of persons, pedestrian misdetections and false alarms sti…

Cited by 0SourceScholar
2018

An End-To-End Siamese Convolutional Neural Network for Loop Closure Detection in Visual Slam System

ICASSP 2018accepted

Loop closure detection is essential and important in visual simultaneous localization and mapping (SLAM) systems. Most existing methods typically utilize a separate feature extraction part and a similarity metric part. Compared to these methods, an end-to-end network is proposed in this paper to joi…

Cited by 0SourceScholar
2016

Communication-efficient weighted ADMM for decentralized network optimization

ICASSP 2016accepted

In this paper, we propose a weighted alternating direction method of multipliers (ADMM) to solve the consensus optimization problem over a decentralized network. Compared with the conventional ADMM that is popular in decentralized network optimization, the weighted ADMM is able to tune its weight ma…

Cited by 0SourceScholar
2015

A proximal gradient algorithm for decentralized nondifferentiable optimization

ICASSP 2015accepted

In this paper, we focus on solving the decentralized consensus optimization problem defined over a networked multi-agent system. All the agents shall cooperatively find a common minimizer of the overall objective while each agent holds its own local objective and can only communicate with its neighb…

Cited by 0SourceScholar