OpenRetina

Open-source repository containing neural network models of the retina. The models in this repository are inspired by and partially contain adapted code of sinzlab/neuralpredictors. Accompanying preprint: openretina: Collaborative Retina Modelling Across Datasets and Species.

Installation

For development and to have access to Jupyter notebooks: git clone git@github.com:open-retina/open-retina.git cd open-retina pip install -e .

For normal usage:

pip install openretina

Test openretina by downloading a model and running a forward pass: ```python import torch from openretina.models import *

model = loadcorereadoutfromremote("hoefling2024baselowres", "cpu") responses = model.forward(torch.rand(model.stimulusshape(timesteps=50))) ```

Contributing

Before raising a PR please run: ```

Fix formatting of python files

make fix-formatting

Run type checks and unit tests

make test-all ```

Design decisions and structure

With this repository we provide already pre-trained retina models that can be used for inference and intepretability out of the box, and dataloaders together with model architectures to train new models. For training new models, we rely on pytorch lightning in combination with hydra to manage the configurations for training and dataloading.

The openretina package is structured as follows: - modules: pytorch modules that define layers and losses - models: pytorch lightning models that define models that can be trained and evaluated (i.e. models from specific papers) - dataio: dataloaders to manage access of data to be used for training - insilico: Methods perform _insilico experiments with above models - stimulus_optimization: optimize inputs for neurons of above models according to interpretable objectives (e.g. most exciting inputs) - future options: gradient analysis, data analysis - utils: Utility functions that are used across above submodules

Related papers and data sources

The Core + Readout model was developed in the paper A chromatic feature detector in the retina signals visual context changes. All datasets used in openretina are shared under a CC-BY Share-Alike license, and we acknowledge and credit the original sources below: - hoefling2024: Originally published by Höfling et al. (2024), eLife - Paper: A chromatic feature detector in the retina signals visual context changes. - Dataset originally deposited at: https://gin.g-node.org/eulerlab/rgc-natstim - karamanlis2024: Originally published by Karamanlis et al. (2024), Nature. - Paper: Nonlinear receptive fields evoke redundant retinal coding of natural scenes - Dataset: Karamanlis D, Gollisch T (2023) Dataset - Marmoset and mouse retinal ganglion cell responses to natural stimuli and supporting data. G-Node. https://doi.org/10.12751/g-node.ejk8kx - maheswaranathan_2023: Originally published by Maheswaranathan et al. (2023), Neuron - Paper: Interpreting the retinal neural code for natural scenes: From computations to neurons - Dataset: Maheswaranathan, N., McIntosh, L., Tanaka, H., Grant, S., Kastner, D., Melander, J., Nayebi, A., Brezovec, L., Wang, J. Ganguli, S. Baccus, S. (2023). Interpreting the retinal neural code for natural scenes: from computations to neurons. Stanford Digital Repository. Available at https://purl.stanford.edu/rk663dm5577

The paper Most discriminative stimuli for functional cell type clustering explains the discriminatory stimulus objective we showcase in notebooks/mostdiscriminativestimulus.