Publications

2025

Abe, E., Zhang, C., Chhibber, R., Chou, G., Foat, J., Truong, D., Olveczky, B., Sniadecki, N., Tuthill, J., Pereira, T., & Brunton, B. (2025). Deep imitation learning for neuromechanical control: Realistic walking in an embodied fly. COSYNE, Montreal, QC.

Aharon, L., Lee, K., Sikka, K., Chettih, S., Hurwitz, C., Paninski, L., & Whiteway, M. R. (2025). An uncertainty-aware framework for data-efficient multi-view animal pose estimation (arXiv:2510.09903). arXiv. https://doi.org/10.48550/arXiv.2510.09903

Altabaa, A., Montasser, O., & Lafferty, J. (2025). CoT Information: Improved Sample Complexity under Chain-of-Thought Supervision (arXiv:2505.15927). arXiv. https://doi.org/10.48550/arXiv.2505.15927

Ananthram, A., Pantalony, R., Bradford, L., Demarest, J., Purvis, A., Krut, K., Stein, R., & McKeown, K. (2025, April). A Benchmark and a Metric for Granular and Coarse Detailed Caption Evaluation.

Ananthram, A., Stengel-Eskin, E., Vondrick, C., Bansal, M., & McKeown, K. (2025, April). Try Seeing It from My Perspective: Characterizing Western Bias in Image Understanding in Large Multimodal Models. ICLR 2025, Singapore.

Bencomo, G., Gupta, M., Marinescu, I., McCoy, R. T., & Griffiths, T. L. (2025). Teasing Apart Architecture and Initial Weights as Sources of Inductive Bias in Neural Networks (arXiv:2502.20237). arXiv. https://doi.org/10.48550/arXiv.2502.20237

Chiquier, M., Avrech, O., Gandelsman, Y., Feng, B., Bouman, K., & Vondrick, C. (2025, April). Teaching Humans Subtle Differences with DIFFusion. arXiv. https://doi.org/10.48550/arXiv.2504.08046

Deng, Z., Zollo, T., Eyre, B., Inamdar, A., Madras, D., & Zemel, R. (2025). QuEst: Enhancing Estimates of Quantile-Based Distributional Measures Using Model Predictions. International Conference on Machine Learning (ICML), Vancouver. https://arxiv.org/abs/2507.05220.

Lu, A., Liao, W., Wang, L., Yang, H., & Shi, J. (2025). Artifacts and Attention Sinks: Structured Approximations for Efficient Vision Transformers (No. arXiv:2507.16018). arXiv. https://doi.org/10.48550/arXiv.2507.16018

Mao, J., Rothkopf, C. A., & Stocker, A. A. (2025). Adaptation optimizes sensory encoding for future stimuli. PLOS Computational Biology, 21(1), e1012746. https://doi.org/10.1371/journal.pcbi.1012746

Rooke, S., Choi, S., & Balasubramanian, V. (2025). Patterns of Ocular Dominance are in the Eye of the Beholder. https://doi.org/10.64898/2025.12.14.694226

Siam, M. A., & Fahim, D. (2025). Self-Supervised Learning for SAR Target Recognition with Multi-Task Pretext Training. 1207–1213. https://doi.org/10.1109/SoutheastCon56624.2025.10971440

Siam, M. A., Noor, D. F., Ndoye, M., & Khan, J. F. (2025). Advancing SAR Target Recognition Through Hierarchical Self-Supervised Learning with Multi-Task Pretext Training. Sensors, 26(1), 122. https://doi.org/10.3390/s26010122

Tyulina, N., Yu, Y., Emmanouil, T. A., & Levitan, S. I. (2025). Beyond Face Value: Visual and Auditory Signals in Human and Machine Trust Judgments. Proceedings of the 7th ACM Conference on Conversational User Interfaces, 1–5. https://doi.org/10.1145/3719160.3737630

Wang, J., Zollo, T., Zemel, R., & Namkoong, H. (2025). Adaptive Elicitation of Latent Information Using Natural Language (arXiv:2504.04204). arXiv. https://doi.org/10.48550/arXiv.2504.04204

Wang, Y., Yu, H., Blau, A., Zhang, Y., Laboratory, T. I. B., Paninski, L., Hurwitz, C., & Whiteway, M. (2025). Self-supervised pretraining of vision transformers for animal behavioral analysis and neural encoding (arXiv:2507.09513). arXiv. https://doi.org/10.48550/arXiv.2507.09513

Wang, Z., Morris, G., Derdikman, D., Chaudhari, P., & Balasubramanian, V. (2025). REMI: Reconstructing Episodic Memory During Internally Driven Path Planning. https://doi.org/10.1101/2025.07.02.662824

Wang, Z., Tullio, R. W. D., Rooke, S., & Balasubramanian, V. (2025, January). Time Makes Space: Emergence of Place Fields in Networks Encoding Temporally Continuous Sensory Experiences. arXiv. https://doi.org/10.48550/arXiv.2408.05798

Yang, H., Xu, K., Grossberg, M. D., Bai, Y., & Shi, J. (2025). “I Know It When I See It”: Mood Spaces for Connecting and Expressing Visual Concepts (No. arXiv:2504.15145). arXiv. https://doi.org/10.48550/arXiv.2504.15145

Zhang, C., Abe, E., Foat, J., Burton, B., Pereira, T., Olveczky, B., & Warnberg, E. (2025). A GPU-Accelerated Deep Reinforcement Learning Pipeline for Simulating Animal Behavior. COSYNE, Montreal.

Zhang, C. Y., Yang, Y., Sirbu, A., Abe, E. T. T., Wärnberg, E., Leonardis, E. J., Aldarondo, D. E., Lee, A., Prasad, A., Foat, J., Bian, K., Park, J., Bhatt, R., Saunders, H., Nagamori, A., Thanawalla, A. R., Huang, K. W., Plum, F., Beck, H. K., … Pereira, T. D. (2025). MIMIC-MJX: Neuromechanical Emulation of Animal Behavior (arXiv:2511.20532). arXiv. https://doi.org/10.48550/arXiv.2511.20532

Zhang, Y., Wang, Y., Azabou, M., Andre, A., Wang, Z., Lyu, H., Laboratory, T. I. B., Dyer, E., Paninski, L., & Hurwitz, C. (2025). Neural Encoding and Decoding at Scale (arXiv:2504.08201). arXiv. https://doi.org/10.48550/arXiv.2504.08201

Zolfaghari, H., Ebrahimi, N., Ji, Y., Pitkow, X., & Davoodi, M. (2025). Integrated Analytical Modeling and Numerical Simulation Framework for Design Optimization of Electromagnetic Soft Actuators. Actuators, 14(3), 128. https://doi.org/10.3390/act14030128

Zolfaghari, H., Ebrahimi, N., Pitkow, X., & Davoodi, M. (2025). Actuator Selection and Control of an Array of Electromagnetic Soft Actuators. Electronics, 14(18), 3682. https://doi.org/10.3390/electronics14183682

2024

Altabaa, A., & Lafferty, J. (2024a). Approximation of relation functions and attention mechanisms (arXiv:2402.08856). arXiv. http://arxiv.org/abs/2402.08856

Altabaa, A., & Lafferty, J. (2024b). Learning Hierarchical Relational Representations through Relational Convolutions (arXiv:2310.03240). arXiv. http://arxiv.org/abs/2310.03240

Altabaa, A., & Lafferty, J. (2025). Disentangling and Integrating Relational and Sensory Information in Transformer Architectures (arXiv:2405.16727). arXiv. https://doi.org/10.48550/arXiv.2405.16727

Azabou, M., Pan, K. X., Arora, V., Knight, I. J., Dyer, E. L., & Richards, B. A. (2024, October). Multi-session, multi-task neural decoding from distinct cell-types and brain regions. https://openreview.net/forum?id=IuU0wcO0mo

Chiquier, M., Mall, U., & Vondrick, C. (2024). Evolving Interpretable Visual Classifiers with Large Language Models. https://doi.org/10.48550/ARXIV.2404.09941

Jesson, A., Beltran-Velez, N., & Blei, D. (2024). Can Generative AI Solve Your In-Context Learning Problem? A Martingale Perspective (arXiv:2412.06033). arXiv. https://doi.org/10.48550/arXiv.2412.06033

Jesson, A., Beltran-Velez, N., & Blei, D. (2024). Estimating the Hallucination Rate of Generative AI (arXiv:2406.07457). arXiv. https://doi.org/10.48550/arXiv.2406.07457

Liang, J., Liu, R., Ozguroglu, E., Sudhakar, S., Dave, A., Tokmakov, P., Song, S., & Vondrick, C. (2024). Dreamitate: Real-World Visuomotor Policy Learning via Video Generation (arXiv:2406.16862). arXiv. https://doi.org/10.48550/arXiv.2406.16862

Mahdaviyeh, Y., Lucas, J., Ren. M., Tolias, A., Zemel, R., Pitassi,. T. (2024). Replay Can Probably Increase Forgetting. NeurIPS. https://arxiv.org/abs/2506.04377.

Olivos-Castillo, I., Schrater, P., & Pitkow, X. (2024, October). Control when confidence is costly. arXiv. https://doi.org/10.48550/arXiv.2406.14427

Ramesh, R., Bisulco, A., DiTullio, R. W., Wei, L., Balasubramanian, V., Daniilidis, K., & Chaudhari, P. (2024, July). Many Perception Tasks are Highly Redundant Functions of their Input Data. arXiv. https://doi.org/10.48550/arXiv.2407.13841

Rooke, S., Wang, Z., Tullio, R. W. D., & Balasubramanian, V. (2024, November). Trading Place for Space: Increasing Location Resolution Reduces Contextual Capacity in Hippocampal Codes. https://openreview.net/forum?id=REIK4SZMJt

Tyulina, N., Emmanouil, T. A., & Levitan, S. I. (2024). Understanding Linguistic and Visual Factors that Affect Human Trust Perception of Virtual Agents. ACM Conversational User Interfaces 2024, 1–6. https://doi.org/10.1145/3640794.3665581

Webb, T. W., Frankland, S. M., Altabaa, A., Segert, S., Krishnamurthy, K., Campbell, D., Russin, J., Giallanza, T., O’Reilly, R., Lafferty, J., & Cohen, J. D. (2024). The relational bottleneck as an inductive bias for efficient abstraction. Trends in Cognitive Sciences, 28(9), 829–843. https://doi.org/10.1016/j.tics.2024.04.001

Xu, K., Zhang, L., & Shi, J. (2024a, April). Detecting Image Attribution for Text-to-Image Diffusion Models in RGB and Beyond. arXiv. https://doi.org/10.48550/arXiv.2403.19653

Xu, K., Zhang, L., & Shi, J. (2024b, May). Good Seed Makes a Good Crop: Discovering Secret Seeds in Text-to-Image Diffusion Models. arXiv. https://doi.org/10.48550/arXiv.2405.14828

Yang, H., Gee, J., & Shi, J. (2024a). Brain Decodes Deep Nets (arXiv:2312.01280). arXiv. http://arxiv.org/abs/2312.01280

Yang, H., Gee, J., & Shi, J. (2024b, June). AlignedCut: Visual Concepts Discovery on Brain-Guided Universal Feature Space. arXiv. https://doi.org/10.48550/arXiv.2406.18344

Zhang, Y., Wang, Y., Jimenez-Beneto, D., Wang, Z., Azabou, M., Richards, B., Winter, O., Laboratory, I. B., Dyer, E., Paninski, L., & Hurwitz, C. (2024). Towards a “universal translator” for neural dynamics at single-cell, single-spike resolution (arXiv:2407.14668). arXiv. https://doi.org/10.48550/arXiv.2407.14668

2023

Altabaa, A., Webb, T., Cohen, J., & Lafferty, J. (2023). Abstractors and relational cross-attention: An inductive bias for explicit relational reasoning in Transformers (arXiv:2304.00195). arXiv. http://arxiv.org/abs/2304.00195

Ding, Z., Tran, D. T., Ponder, K., Cobos, E., Ding, Z., Fahey, P. G., Wang, E., Muhammad, T., Fu, J., Cadena, S. A., Papadopoulos, S., Patel, S., Franke, K., Reimer, J., Sinz, F. H., Ecker, A. S., Pitkow, X., & Tolias, A. S. (2023, March). Bipartite invariance in mouse primary visual cortex. bioRxiv. https://doi.org/10.1101/2023.03.15.532836

Eyre, B., Creager, E., Madras, D., Papyan, V., & Zemel, R. (2023). Out of the Ordinary: Spectrally Adapting Regression for Covariate Shift (arXiv:2312.17463). arXiv. http://arxiv.org/abs/2312.17463

McGaughey, K. D., & Gold, J. (2023). Neuroscience 2023. Contributions of sensory adaptation and pupil-linked arousal to perceptual decisions about uncertain and unstable visual stimuli. Neuroscience 2023.

Under Review