Any explanation on the choice of the colormap? It can have a powerful effect on how we perceive this visualization.
My process is to periodically prompt the model as I fine-tune. The features that seem to correlate with the model losing coherence are highlighted nicely with this color mapping. More of an art than a science.