arXiv Machine Learning By Nico Policzer, James Ho, Joseph Soo, Xin Tang

Neural-DISCO: Source-Conditioned Counterfactual Editing of Neural Population Activity

Read the original on arXiv Machine Learning →

Neural‑DISCO is a disentangled representation learning framework that enables source‑conditioned counterfactual editing of neural population activity. By partitioning latent space into label‑specific components and a residual latent, it allows selective editing of behavioral or stimulus labels while preserving other observed labels and intrinsic variability. The method is validated on simulated and real neural recordings, and combined with feature attribution to identify neurons most relevant to each label, improving recovery of known functional cell roles.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Oct 5

NeuroLens: Learning Latent Embeddings of Neural Semantics from Chronic Recordings

NeuroLens is a self‑supervised model that learns denoised, semantically informative latent embeddings from chronic neural recordings using a Joint‑Embedding Predictive Architecture. It maps changing neural populations into a common latent space and predicts future latent states, thereby reducing sensitivity to transient recording variability. Across chronic intracortical data in mice and humans, NeuroLens improves decoding of decision‑making and semantic task variables, supports multi‑day pretraining, rapid few‑shot adaptation, and yields more stable decoding over time than existing baselines.

By Hanrui Lyu, Baiyuan Chen, Tianshu Tan, Matthew R. Whiteway, Maxwell D. Melin, Ji Xia, Linyang He, Bradly C. Stadie, Anne Churchland, Liam Paninski, Yizi Zhang
arXiv Machine Learning
Oct 23

Interpreting Model Reasoning through Behavioral Counterfactuals

The paper introduces a method for analyzing how interventions in a prompt affect a language model’s reasoning by using behavioral counterfactuals. It trains a module to edit the residual stream so that the next-token distributions for a fixed response match between a source and a counterfactual prompt, outperforming standard activation reconstruction. The resulting attribution maps and visualizations help identify causally important locations for counterfactual recovery and guide hypothesis testing with activation patching, as demonstrated in a case study on Gemma-4-E4B-it and DeepSeek-R1-0528-Qwen3-8B.

By Shawn Im, James Oldfield, Adel Bibi, Sharon Li