I don't know if training would help alone, forcing your brain to create new neuronal connections is probably necessary.

Personally that came from deep painful emotional instances in my early life where I had nowhere else to hide but within myself. I have 3d imagination in which I can create full environment and caracters. It is entirely different from dreaming that I remember while waking up,. Memories or half awake dreams that I make operational decisions.

I actually had the reverse issue of you. I had to learn not to imagine what was discussed, saw or dreamed since some people would discuss things I didn't want to imagine. It made my memory worse though.

https://youtube.com/@codeparade

Since finding this developer, I have been trying to visualize 3D environmental as 4D. I can create some new axial rotation, but I am still trying to determine how to find the actual 4th vectorial rotation in objects. Unusual problem to have I guess.