← Search

Bin Yu

32 accepted papers

2026

LangForce: Bayesian Decomposition of Vision Language Action Models via Latent Action Queries

ICML 2026poster

Vision-Language-Action (VLA) models have shown promise in robot manipulation but often struggle to generalize to new instructions or complex multi-task scenarios. We identify a critical pathology in current training paradigms where goal-driven data collection creates a dataset bias. In such datasets…

Cited by 0SourceScholar
2026

Sampled hard labels from sparse targets mislead rotation invariant algorithms

ICML 2026poster

One of the most common machine learning setups is logistic regression. In many classification models, including neural networks, the final prediction is obtained by applying a logistic link function to a linear score. In binary logistic regression, the feedback can be either soft labels, correspondi…

Cited by 0SourceScholar
2026

T4NMTD: Transition-Centric Reinforcement Learning for Non-Markovian Task Decomposition

AAAI 2026technical

Non-Markovian Tasks (NMTs) are distinguished by their dependence on long-term memory and state-dependent dynamics, setting them apart from the traditional Markovian models typically employed in Reinforcement Learning (RL). NMTs not only suffer from reward sparseness but also rely on historical infor

Cited by 0SourcePDFScholar
2025

Benefits of Early Stopping in Gradient Descent for Overparameterized Logistic Regression

ICML 2025poster

In overparameterized logistic regression, gradient descent (GD) iterates diverge in norm while converging in direction to the maximum $\ell_2$-margin solution---a phenomenon known as the implicit bias of GD. This work investigates additional regularization effects induced by early stopping in well-s…

Cited by 0SourcePDFScholar
2025

Efficient Automated Circuit Discovery in Transformers using Contextual Decomposition

ICLR 2025poster

Automated mechanistic interpretation research has attracted great interest due to its potential to scale explanations of neural network internals to large models. Existing automated circuit discovery work relies on activation patching or its approximations to identify subgraphs in models for specifi…

Cited by 1SourcePDFScholar
2025

MITIGATING OVER-EXPLORATION IN LATENT SPACE OPTIMIZATION USING LES

ICML 2025poster

We develop Latent Exploration Score (LES) to mitigate over-exploration in Latent Space Optimization (LSO), a popular method for solving black-box discrete optimization problems. LSO utilizes continuous optimization within the latent space of a Variational Autoencoder (VAE) and is known to be suscept…

Cited by 0SourcePDFScholar
2025

Proxy-SPEX: Sample-Efficient Interpretability via Sparse Feature Interactions in LLMs

NeurIPS 2025spotlight

Large Language Models (LLMs) have achieved remarkable performance by capturing complex interactions between input features. To identify these interactions, most existing approaches require enumerating all possible combinations of features up to a given order, causing them to scale poorly with the nu…

Cited by 0SourcecodeScholar
2025

SPEX: Scaling Feature Interaction Explanations for LLMs

ICML 2025poster

Large language models (LLMs) have revolutionized machine learning due to their ability to capture complex interactions between input features. Popular post-hoc explanation methods like SHAP provide *marginal* feature attributions, while their extensions to interaction importances only scale to small…

2025

Towards Consistent Natural-Language Explanations via Explanation-Consistency Finetuning

COLING 2025main

Large language models (LLMs) often generate convincing, fluent explanations. However, different from humans, they often generate inconsistent explanations on different inputs. For example, an LLM may explain “all birds can fly” when answering the question “Can sparrows fly?” but meanwhile answer “no…

2024

Diagnosing Transformers: Illuminating Feature Spaces for Clinical Decision-Making

ICLR 2024poster

Pre-trained transformers are often fine-tuned to aid clinical decision-making using limited clinical notes. Model interpretability is crucial, especially in high-stakes domains like medicine, to establish trust and ensure safety, which requires human engagement. We introduce SUFO, a systematic frame…

2024

ED-Copilot: Reduce Emergency Department Wait Time with Language Model Diagnostic Assistance

ICML 2024poster

In the emergency department (ED), patients undergo triage and multiple laboratory tests before diagnosis. This time-consuming process causes ED crowding which impacts patient mortality, medical errors, staff burnout, etc. This work proposes (time) *cost-effective diagnostic assistance* that leverage…

2024

Improving Prototypical Visual Explanations with Reward Reweighing, Reselection, and Retraining

ICML 2024poster

In recent years, work has gone into developing deep interpretable methods for image classification that clearly attributes a model's output to specific features of the data. One such of these methods is the Prototypical Part Network (ProtoPNet), which attempts to classify images based on meaningful…

2024

Tell Your Model Where to Attend: Post-hoc Attention Steering for LLMs

ICLR 2024poster

In human-written articles, we often leverage the subtleties of text style, such as bold and italics, to guide the attention of readers. These textual emphases are vital for the readers to grasp the conveyed information. When interacting with large language models (LLMs), we have a similar need -- s…

2023

Bridging Discrete and Backpropagation: Straight-Through and Beyond

NeurIPS 2023oral

Backpropagation, the cornerstone of deep learning, is limited to computing gradients for continuous variables. This limitation poses challenges for problems involving discrete latent variables. To address this issue, we propose a novel approach to approximate the gradient of parameters involved in g…

2022

A cautionary tale on fitting decision trees to data from additive models: generalization lower bounds

AISTATS 2022poster

Decision trees are important both as interpretable models amenable to high-stakes decision-making, and as building blocks of ensemble methods such as random forests and gradient boosting. Their statistical properties, however, are not well understood. The most cited prior works have focused on deriv…

2022

C2AM Loss: Chasing a Better Decision Boundary for Long-Tail Object Detection

CVPR 2022poster

Long-tail object detection suffers from poor performance on tail categories. We reveal that the real culprit lies in the extremely imbalanced distribution of the classifier's weight norm. For conventional softmax cross-entropy loss, such imbalanced weight norm distribution yields ill conditioned dec…

Cited by 28PDFScholar
2022

Hierarchical Shrinkage: Improving the accuracy and interpretability of tree-based models.

ICML 2022oral

Decision trees and random forests (RF) are a cornerstone of modern machine learning practice. Due to their tendency to overfit, trees are typically regularized by a variety of techniques that modify their structure (e.g. pruning). We introduce Hierarchical Shrinkage (HS), a post-hoc algorithm which…

2021

Adaptive wavelet distillation from neural networks through interpretations

NeurIPS 2021poster

Recent deep-learning models have achieved impressive prediction performance, but often sacrifice interpretability and computational efficiency. Interpretability is crucial in many disciplines, such as science and medicine, where models must be carefully vetted or where interpretation is the goal its…

Cited by 51SourcePDFScholar
2021

High-Performance Discriminative Tracking With Transformers

ICCV 2021poster

End-to-end discriminative trackers improve the state of the art significantly, yet the improvement in robustness and efficiency is restricted by the conventional discriminative model, i.e., least-squares based regression. In this paper, we present DTT, a novel single-object discriminative tracker, b…

Cited by 141PDFScholar
2020

Interpretations are Useful: Penalizing Explanations to Align Neural Networks with Prior Knowledge

ICML 2020poster

For an explanation of a deep learning model to be effective, it must provide both insight into a model and suggest a corresponding action in order to achieve some objective. Too often, the litany of proposed explainable deep learning methods stop at the first step, providing practitioners with insig…

2020

Sharp Analysis of Expectation-Maximization for Weakly Identifiable Models

AISTATS 2020poster

We study a class of weakly identifiable location-scale mixture models for which the maximum likelihood estimates based on $n$ i.i.d. samples are known to have lower accuracy than the classical $n^{- \frac{1}{2}}$ error. We investigate whether the Expectation-Maximization (EM) algorithm also converge…

Cited by 33SourcePDFScholar
2019

A Debiased MDI Feature Importance Measure for Random Forests

NeurIPS 2019poster

Tree ensembles such as Random Forests have achieved impressive empirical success across a wide variety of applications. To understand how these models make predictions, people routinely turn to feature importance measures calculated from tree ensembles. It has long been known that Mean Decrease Impu…

2018

Beyond Word Importance: Contextual Decomposition to Extract Interactions from LSTMs

ICLR 2018oral

The driving force behind the recent success of LSTMs has been their ability to learn complex and non-linear relationships. Consequently, our inability to describe these relationships has led to LSTMs being characterized as black boxes. To this end, we introduce contextual decomposition (CD), an inte…

2016

Supervised Neighborhoods for Distributed Nonparametric Regression

AISTATS 2016poster

Techniques for nonparametric regression based on fitting small-scale local models at prediction time have long been studied in statistics and pattern recognition, but have received less attention in modern large-scale machine learning applications. In practice, such methods are generally applied to…

Cited by 38SourcePDFScholar