← Search

Bowei He

24 accepted papers

2026

Distributionally Robust Set Representation Learning Under Inference-Time Element Corruption

ICML 2026poster

Standard Set Representation Learning methods typically excel on curated data but often overlook the challenge of Inference-time Element Corruption. This refers to scenarios where deployed models encounter element-level degradations, such as outliers or missing components, that may distort the set re…

Cited by 0SourceScholar
2026

Less Is More: Elevating RAG via Performance-Driven Context Compression

ICML 2026poster

Retrieval-Augmented Generation (RAG) has emerged as a promising paradigm for improving the timeliness of knowledge updates and the factual accuracy of large language models. However, incorporating a large volume of retrieved documents significantly increases input length, leading to prohibitive comp…

Cited by 0SourceScholar
2026

PASER: Post-Training Data Selection for Efficient Pruned Large Language Model Recovery

ICLR 2026poster

Model pruning is an effective approach for compressing large language models (LLMs). However, this process often leads to significant degradation of model capabilities. While post-training techniques such as instruction tuning are commonly employed to recover model performance, existing methods ofte…

Cited by 0SourceScholar
2026

Pedagogically-Inspired Data Synthesis for Language Model Knowledge Distillation

ICLR 2026poster

Knowledge distillation from Large Language Models (LLMs) to smaller models has emerged as a critical technique for deploying efficient AI systems. However, current methods for distillation via synthetic data lack pedagogical awareness, treating knowledge transfer as a one-off data synthesis and trai…

Cited by 0SourceScholar
2026

RECODE: A Benchmark for Research Code DEvelopment with Interactive Human Feedback

ICLR 2026poster

Large language models (LLMs) show the promise in supporting scientific research implementation, yet their ability to generate correct and executable code remains limited. Existing works largely adopt one-shot settings, ignoring the iterative and feedback-driven nature of realistic workflows of scien…

Cited by 0SourcecodeScholar
2026

Search-R2: Enhancing Search-Integrated Reasoning via Actor-Refiner Collaboration

ICML 2026poster

Search-integrated reasoning enables language agents to transcend static parametric knowledge by actively querying external sources. However, training these agents via reinforcement learning is hindered by the *multi-scale credit assignment* problem: existing methods typically rely on sparse, traject…

Cited by 0SourceScholar
2026

Set Representation Auxiliary Learning with Adversarial Encoding Perturbation and Optimization

ICLR 2026poster

Sets are a fundamental data structure, and learning their vectorized representations is crucial for many computational problems. Existing methods typically focus on intra-set properties such as permutation invariance and cardinality independence. While effective at preserving basic intra-set semanti…

Cited by 0SourceScholar
2026

Spatial CAPTCHA: Generatively Benchmarking Spatial Reasoning for Human-Machine Differentiation

ICLR 2026poster

Online services rely on CAPTCHAs as a first line of defense against automated abuse, yet recent advances in multi-modal large language models (MLLMs) have eroded the effectiveness of conventional designs that focus on text recognition or 2D image understanding. To address this challenge, we present…

Cited by 0SourcecodeScholar
2026

Support-Proximity Augmented Diffusion Estimation for Offline Black-Box Optimization

ICML 2026poster

Offline black-box optimization aims to discover novel designs with high property scores using only a static dataset, a task fundamentally challenged by the out-of-distribution (OOD) extrapolation problem. Existing approaches typically bifurcate into inverse methods, which struggle with the ill-posed…

Cited by 0SourceScholar
2026

S²Drug: Bridging Protein Sequence and 3D Structure in Contrastive Representation Learning for Virtual Screening

AAAI 2026technical

Virtual screening (VS) is an essential task in drug discovery, focusing on the identification of small-molecule ligands that bind to specific protein pockets. Existing deep learning methods, from early regression models to recent contrastive learning approaches, primarily rely on structural data whi

Cited by 0SourcePDFScholar
2025

CIDD: Collaborative Intelligence for Structure-Based Drug Design Empowered by LLMs

NeurIPS 2025poster

Structure-guided molecular generation is pivotal in early-stage drug discovery, enabling the design of compounds tailored to specific protein targets. However, despite recent advances in 3D generative modeling, particularly in improving docking scores, these methods often produce rare and intrinsica…

Cited by 0SourceScholar
2025

Certifying Language Model Robustness with Fuzzed Randomized Smoothing: An Efficient Defense Against Backdoor Attacks

ICLR 2025poster

The widespread deployment of pre-trained language models (PLMs) has exposed them to textual backdoor attacks, particularly those planted during the pre-training stage. These attacks pose significant risks to high-reliability applications, as they can stealthily affect multiple downstream tasks. Whil…

Cited by 0SourcePDFScholar
2025

NILE: Internal Consistency Alignment in Large Language Models

EMNLP 2025

Recent advances show that the world knowledge in the Instruction Fine-Tuning (IFT) dataset, which is incompatible with LLMs’ internal knowledge, can greatly hurt the IFT performance. However, the effective integration and balancing of the internal knowledge of LLMs, acquired during pre-training, wit

2025

Preserving LLM Capabilities through Calibration Data Curation: From Analysis to Optimization

NeurIPS 2025poster

Post-training compression has been a widely employed approach to scale down large language model (LLM) and facilitate efficient inference. In various proposed compression methods, including pruning and quantization, calibration data plays a vital role by informing the weight importance and activatio…

Cited by 0SourcecodeScholar
2025

SRA-CL: Semantic Retrieval Augmented Contrastive Learning for Sequential Recommendation

NeurIPS 2025poster

Contrastive learning has shown effectiveness in improving sequential recommendation models. However, existing methods still face challenges in generating high-quality contrastive pairs: they either rely on random perturbations that corrupt user preference patterns or depend on sparse collaborative d…

Cited by 0SourcecodeScholar
2025

Sens-Merging: Sensitivity-Guided Parameter Balancing for Merging Large Language Models

ACL 2025finding

Recent advances in large language models have led to numerous task-specialized fine-tuned variants, creating a need for efficient model merging techniques that preserve specialized capabilities while avoiding costly retraining. While existing task vector-based merging methods show promise, they typi…

2024

Bi-Chainer: Automated Large Language Models Reasoning with Bidirectional Chaining

ACL 2024findings

Large Language Models (LLMs) have shown human-like reasoning abilities but still face challenges in solving complex logical problems. Existing unidirectional chaining methods, such as forward chaining and backward chaining, suffer from issues like low prediction accuracy and efficiency. To address t…

2024

No Time to Train: Empowering Non-Parametric Networks for Few-shot 3D Scene Segmentation

CVPR 2024highlight

To reduce the reliance on large-scale datasets recent works in 3D segmentation resort to few-shot learning. Current 3D few-shot segmentation methods first pre-train models on 'seen' classes and then evaluate their generalization performance on 'unseen' classes. However the prior pre-training stage n…

2023

MMEL: A Joint Learning Framework for Multi-Mention Entity Linking

UAI 2023poster

Entity linking, bridging mentions in the contexts with their corresponding entities in the knowledge bases, has attracted wide attention due to many potential applications. Recently, plenty of multimodal entity linking approaches have been proposed to take full advantage of the visual information ra…

2023

Mutually Guided Few-Shot Learning For Relational Triple Extraction

ICASSP 2023accepted

Knowledge graphs (KGs), containing many entity-relation-entity triples, provide rich information for downstream applications. Although extracting triples from unstructured texts has been widely explored, most of them require a large number of labeled instances. The performance will drop dramatically…

Cited by 0SourceScholar
2023

Not All Features Matter: Enhancing Few-shot CLIP with Adaptive Prior Refinement

ICCV 2023poster

The popularity of Contrastive Language-Image Pre-training (CLIP) has propelled its application to diverse downstream vision tasks. To improve its capacity on downstream tasks, few-shot learning has become a widely-adopted technique. However, existing methods either exhibit limited performance or suf…

Cited by 89PDFcodeScholar
2023

Offline Imitation Learning with Variational Counterfactual Reasoning

NeurIPS 2023poster

In offline imitation learning (IL), an agent aims to learn an optimal expert behavior policy without additional online environment interactions. However, in many real-world scenarios, such as robotics manipulation, the offline dataset is collected from suboptimal behaviors without rewards. Due to th…

2023

PointCLIP V2: Prompting CLIP and GPT for Powerful 3D Open-world Learning

ICCV 2023poster

Large-scale pre-trained models have shown promising open-world performance for both vision and language tasks. However, their transferred capacity on 3D point clouds is still limited and only constrained to the classification task. In this paper, we first collaborate CLIP and GPT to be a unified 3D…

Cited by 241PDFcodeScholar
2022

Collective Conditioned Reflex: A Bio-Inspired Fast Emergency Reaction Mechanism for Designing Safe Multi-Robot Systems

RA-L 2022

A multi-robot system (MRS) is a group of coordinated robots designed to cooperate with each other and accomplish given tasks. Due to the uncertainties in operating environments, the system may encounter emergencies, such as unobserved obstacles, moving vehicles, and extreme weather. Animal groups su

Cited by 5SourceScholar