Update README.md

README.md

@@ -21,17 +21,104 @@ This repository contains the **ImageNet-1K SimCLR ResNet-18 checkpoint** trained
 
 **Diaz, D. M., & Henderson, M. M. (2026). _Eccentricity-Constrained CNN Training Reveals Adaptive Information Coding Around the Visual Field._ Proceedings of the 9th Conference on Cognitive Computational Neuroscience.**
 
-**DOI:** `10.32470/0416gfsq`<br>
-**arXiv:** `2607.19316`<br>
+**DOI:** [10.32470/0416gfsq](https://doi.org/10.32470/0416gfsq)<br>
+**arXiv:** [2607.19316](https://arxiv.org/abs/2607.19316)<br>
 **Contributed Talk:** [CCN 2026 presentation on YouTube](https://www.youtube.com/watch?v=Lb4S3FWqd2M&t=2545s)
 
-The model was pretrained using **SimCLR with a ResNet-18 backbone** and served as one of the non-egocentric reference models in the associated study. It was used alongside models pretrained on ImageNet-100 and STL-10 to provide comparison points for evaluating representations learned from naturalistic egocentric visual experience.
+The model was pretrained using **SimCLR with a ResNet-18 backbone** and served as one of the non-egocentric reference models in the associated study. It was evaluated alongside models pretrained on ImageNet-100 and STL-10 as comparison models for representations learned from naturalistic egocentric visual experience.
 
-Training was implemented using the [**Lightly** self-supervised learning framework](https://docs.lightly.ai/self-supervised-learning/index.html). The training images were obtained from the [`evanarlian/imagenet_1k_resized_256`](https://huggingface.co/datasets/evanarlian/imagenet_1k_resized_256) dataset on Hugging Face.
+Training was implemented using the [Lightly self-supervised learning framework](https://docs.lightly.ai/self-supervised-learning/index.html). The training images were obtained from the [`evanarlian/imagenet_1k_resized_256`](https://huggingface.co/datasets/evanarlian/imagenet_1k_resized_256) dataset on Hugging Face.
+
+## Model Architecture
+
+The model uses a standard **ResNet-18** encoder with the classification head removed.
+
+| Component | Configuration |
+| --- | --- |
+| Backbone | ResNet-18 |
+| Backbone representation | 512 dimensions |
+| Projection head | Lightly `SimCLRProjectionHead` |
+| Projection dimensions | `512 → 512 → 128` |
+| Projection output | 128 dimensions |
+| SSL objective | NT-Xent |
+| Temperature | `0.1` |
+
+The released checkpoint contains both the ResNet-18 backbone and SimCLR projection head. For downstream representation extraction, the 512-dimensional backbone representation can be used independently of the projection head.
+
+## Training Configuration
+
+| Parameter | Value |
+| --- | --- |
+| Dataset | ImageNet-1K |
+| Dataset source | `evanarlian/imagenet_1k_resized_256` |
+| Number of classes | 1,000 |
+| Epochs | 100 |
+| Batch size | 32 |
+| Input resolution | `224 × 224` |
+| Optimizer | LARS |
+| Initial learning rate | `0.0375` |
+| Momentum | `0.9` |
+| Weight decay | `1e-6` |
+| LR schedule | Cosine warmup |
+| Warmup | 10 epochs |
+| Precision | 16-bit mixed precision |
+| Distributed training | No |
+
+The learning rate was linearly scaled from a base learning rate of `0.3` according to batch size:
+
+`0.3 × (32 / 256) = 0.0375`
+
+The training script specifies 100 epochs, 1,000 classes, and 224-pixel inputs. The checkpoint was saved at the completion of this run.
+
+## Training Data
+
+Training data were obtained from the Hugging Face dataset:
+
+[`evanarlian/imagenet_1k_resized_256`](https://huggingface.co/datasets/evanarlian/imagenet_1k_resized_256)
+
+The locally downloaded dataset was loaded from Hugging Face parquet shards. The training split was used for self-supervised representation learning.
+
+The dataset itself is **not redistributed through this repository** and remains subject to its original access conditions and terms.
+
+## Checkpoint
+
+**File:** `checkpoint_100-resnet18-simclr-imagenet1k.ckpt`
+
+The released file is a **full PyTorch Lightning checkpoint**, rather than a backbone-only state dictionary.
+
+Checkpoint inspection confirmed:
+
+| Property | Value |
+| --- | --- |
+| PyTorch Lightning version recorded | `2.6.1` |
+| Stored epoch | `99` |
+| Training epochs completed | 100 |
+| Global step | `4,003,600` |
+| State-dict entries | 132 |
+| Backbone output | 512 dimensions |
+| Projection output | 128 dimensions |
+| Strict architecture loading | Successful |
+
+The stored epoch is zero-indexed, so `epoch = 99` corresponds to the completion of epoch 100.
+
+The checkpoint includes training state such as optimizer and scheduler information in addition to model parameters.
+
+## Intended Use
+
+This checkpoint is provided for research and downstream applications involving self-supervised visual representations, including:
+
+- reproducing the reference-model analyses reported in Diaz and Henderson (2026),
+- extracting ResNet-18 representations for comparison with the VEDB-pretrained models,
+- reproducing the associated NSD voxelwise encoding analyses,
+- linear-probe or fine-tuned image classification,
+- transfer learning to other visual recognition tasks, and
+- representation-learning and visual-neuroscience research.
+
+The released checkpoint contains a self-supervised ResNet-18 encoder and SimCLR projection head rather than a trained classification head. For image classification, users can attach and train an appropriate classifier on the learned backbone representations or fine-tune the encoder for the target task.
 
 ## Related Models
 
-This repository is associated with the **[Eccentricity-Constrained SimCLR Models (VEDB)](https://hf.co/collections/DM-Diaz/eccentricity-constrained-simclr-models-vedb)** collection.
+This model was used as a non-egocentric reference model in the study associated with the **[Eccentricity-Constrained SimCLR Models (VEDB)](https://hf.co/collections/DM-Diaz/eccentricity-constrained-simclr-models-vedb)** collection.
 
 - [VEDB SimCLR ResNet-18 — Baseline](https://huggingface.co/DM-Diaz/VEDB-SimCLR-ResNet18-Baseline)
 - [VEDB SimCLR ResNet-18 — Fovea-Gaze](https://huggingface.co/DM-Diaz/VEDB-SimCLR-ResNet18-Fovea-Gaze)
@@ -40,4 +127,18 @@ This repository is associated with the **[Eccentricity-Constrained SimCLR Models
 - [VEDB NSD ResNet-18 — Encoding Models](https://huggingface.co/DM-Diaz/VEDB-NSD-ResNet18-Encoding-Models)
 - [SimCLR ResNet-18 — ImageNet-1K](https://huggingface.co/DM-Diaz/SimCLR-ResNet18-ImageNet1K)
 - [SimCLR ResNet-18 — ImageNet-100](https://huggingface.co/DM-Diaz/SimCLR-ResNet18-ImageNet100)
-- [SimCLR ResNet-18 — STL-10](https://github.com/Spijkervet/SimCLR) *(external pretrained reference model; checkpoint provided by Spijkervet/SimCLR and not redistributed by this project)*
+- [SimCLR ResNet-18 — STL-10](https://github.com/Spijkervet/SimCLR) *(external pretrained reference model; checkpoint provided by Spijkervet/SimCLR and not redistributed by this project)*
+
+## Citation
+
+If you use this checkpoint or representations derived from it in academic work, please cite the associated study:
+
+```bibtex
+@inproceedings{diaz2026eccentricity,
+  author    = {Diaz, Dylan M. and Henderson, Margaret M.},
+  title     = {Eccentricity-Constrained CNN Training Reveals Adaptive Information Coding Around the Visual Field},
+  booktitle = {Proceedings of the 9th Conference on Cognitive Computational Neuroscience},
+  address   = {New York, NY, USA},
+  year      = {2026},
+  doi       = {10.32470/0416gfsq}
+}