We should extend the explainability tools developed for robust virtual staining to representation learning. I suggest the following:
- Export a movie of features (input, PC of features) for a chosen track.
- Effective receptive field of a chosen embedding dimension.