SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2505.24048 · ICML · 2025

NeuronTune: Towards Self-Guided Spurious Bias Mitigation

Guangtao Zheng, Wenqian Ye, Aidong Zhang

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 12 functions out of this paper's own repositories and ran 8 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
gtzheng/neurontune canonical 8 of 12
FunctionStatusWhere it lives
ClassBalancedSampler Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("f0e41e9e56aec872")
EmbedDataset Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("a0e653064ea9bb28")
GroupBalancedSampler Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("21f32fcbac2781e6")
JointRandomSampler Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("7b44ba5417ec1392")
MaskLayer Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("b972838e4feca157")
RandomSampler Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("01aaa3628e4c6a24")
ShortcutSampler Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("a57a74a821996c30")
extract_feature_info Ran gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("92b6651ceba09ac2")
Algorithm Not yet run gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("7a4ce898b756fc69")
MaskedClassifier Not yet run gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("3834a31863be3927")
NeuronTune Not yet run gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("27146a31d6d6752d")
prepare_data Not yet run gtzheng/neurontune/algorithms/neuron_tune.py
pointer only (licence: NONE) · get_code("ef12d101ba623542")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Deep neural networks often develop spurious bias, reliance on correlations between non-essential features and classes for predictions. For example, a model may identify objects based on frequently co-occurring backgrounds rather than intrinsic features, resulting in degraded performance on data lacking these correlations. Existing mitigation approaches typically depend on external annotations of spurious correlations, which may be difficult to obtain and are not relevant to the spurious bias in a model. In this paper, we take a step towards self-guided mitigation of spurious bias by proposing NeuronTune, a post hoc method that directly intervenes in a model's internal decision process. Our method probes in a model's latent embedding space to identify and regulate neurons that lead to spurious prediction behaviors. We theoretically justify our approach and show that it brings the model closer to an unbiased one. Unlike previous methods, NeuronTune operates without requiring spurious correlation annotations, making it a practical and effective tool for improving model robustness. Experiments across different architectures and data modalities demonstrate that our method significantly mitigates spurious bias in a self-guided way.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2505.24048")
get_code_for_paper("2505.24048")
have("2505.24048")

Connect an agent — have() is free.