← Search

Yiwei Chen

17 accepted papers

2026

Flow-Based Knowledge Transfer for Efficient Large Model Distillation

AAAI 2026technical

Traditional knowledge distillation relies on simple MSE or KL divergence losses that fail to capture the complex distributional relationships between teacher and student model representations. We propose FlowDistill, a novel distillation framework that employs normalizing flows to model and transfer

Cited by 0SourcePDFScholar
2026

Leak@$k$: Unlearning Does Not Make LLMs Forget Under Probabilistic Decoding

ICML 2026poster

Unlearning in large language models (LLMs) is critical for regulatory compliance and for building ethical generative AI systems that avoid producing private, toxic, illegal, or copyrighted content. Despite rapid progress, in this work we show that \textit{almost all} existing unlearning methods fail…

Cited by 0SourceScholar
2026

Less Is More: Fast and Accurate Reasoning with Cross-Head Unified Sparse Attention

ICML 2026poster

Large reasoning models achieve strong performance through test-time scaling, but this incurs substantial computational overhead due to long decoding from short prompts. While sparse attention can reduce latency and memory usage, existing methods often degrade reasoning accuracy because selection err…

Cited by 0SourceScholar
2026

MaskGuide: Efficient Distillation for Deployable Lightweight Segmentation in Marine Environments

RA-L 2026

The growing demand for efficient image segmentation in marine ecological studies is currently constrained by two key factors: the high computational requirements of models such as the Segment Anything Model (SAM) and the degraded accuracy of lightweight models in underwater environments. To overcome

Cited by 0SourceScholar
2026

Safety Mirage: How Spurious Correlations Undermine VLM Safety Fine-Tuning and Can Be Mitigated by Machine Unlearning

ICLR 2026poster

Recent vision language models (VLMs) have made remarkable strides in generative modeling with multimodal inputs, particularly text and images. However, their susceptibility to generating harmful content when exposed to unsafe queries raises critical safety concerns. While current alignment strategie…

Cited by 0SourcecodeScholar
2026

Unlearning Isn't Invisible: Detecting Unlearning Traces in LLMs from Model Outputs

ICLR 2026poster

Machine unlearning (MU) for large language models (LLMs), commonly referred to as LLM unlearning, seeks to remove specific undesirable data or knowledge from a trained model, while maintaining its performance on standard tasks. While unlearning plays a vital role in protecting data privacy, enforcin…

Cited by 0SourcecodeScholar
2026

Zephyrus: An Agentic Framework for Weather Science

ICLR 2026poster

Foundation models for weather science are pre-trained on vast amounts of structured numerical data and outperform traditional weather forecasting systems. However, these models lack language-based reasoning capabilities, limiting their utility in interactive scientific workflows. Large language mode…

Cited by 0SourcecodeScholar
2025

Pushing Wi-Fi Towards Fine-Grained Sensing Via Spectrogram Enhancement

ICASSP 2025accepted

In recent years, Wi-Fi sensing has attracted much attention due to the widespread deployment of communication devices. Due to advancements in signal processing algorithms, contactless sensing technology based on Wi-Fi signals has now been widely applied. However, the limited bandwidth of Wi-Fi syste…

Cited by 0SourceScholar
2025

Retrospective Learning from Interactions

ACL 2025long

Multi-turn interactions between large language models (LLMs) and users naturally include implicit feedback signals. If an LLM responds in an unexpected way to an instruction, the user is likely to signal it by rephrasing the request, expressing frustration, or pivoting to an alternative task. Such s…

Cited by 0SourcePDFScholar
2025

The Fragile Truth of Saliency: Improving LLM Input Attribution via Attention Bias Optimization

NeurIPS 2025spotlight

Input saliency aims to quantify the influence of input tokens on the output of large language models (LLMs), which has been widely used for prompt engineering, model interpretability, and behavior attribution. Despite the proliferation of saliency techniques, the field lacks a standardized and rigor…

Cited by 0SourceScholar
2024

Acoustic Array Robot Based on the Dual-Rotation Mechatronic Design

RA-L 2024

With the ability to manipulate objects contactlessly, acoustic levitation has excellent potential in the micro-assembly industry. Fixed ultrasonic phased arrays (UPA) widely used in acoustic levitation control objects with restricted space and inflexibility. Here, an acoustic array robot based on th

Cited by 4SourceScholar
2024

Efficient Learning on Successive Test Time Augmentation

ICASSP 2024accepted

Test time augmentation (TTA) has been a promising tool for improving the robustness against out-of-distribution data at inference time. Recent TTA methods try to learn predictive transformations which are supposed to provide the best performance gain on each test sample. However, existing methods ar…

Cited by 0SourceScholar
2024

MarineInst: A Foundation Model for Marine Image Analysis with Instance Visual Description

ECCV 2024oral

"Recent foundation models trained on a tremendous scale of data have shown great promise in a wide range of computer vision tasks and application domains. However, less attention has been paid to the marine realms, which in contrast cover the majority of our blue planet. The scarcity of labeled data…

Cited by 8SourcePDFScholar
2023

BadTrack: A Poison-Only Backdoor Attack on Visual Object Tracking

NeurIPS 2023poster

Visual object tracking (VOT) is one of the most fundamental tasks in computer vision community. State-of-the-art VOT trackers extract positive and negative examples that are used to guide the tracker to distinguish the object from the background. In this paper, we show that this characteristic can b…

Cited by 6SourcePDFScholar
2021

Learning Generalized Intersection Over Union for Dense Pixelwise Prediction

ICML 2021spotlight

Intersection over union (IoU) score, also named Jaccard Index, is one of the most fundamental evaluation methods in machine learning. The original IoU computation cannot provide non-zero gradients and thus cannot be directly optimized by nowadays deep learning methods. Several recent works generaliz…

Cited by 33SourcePDFScholar