Your description of the problem lines up with my experience. Finding multiple replies in a large topic is incredibly tedious and time consuming.
Personally, I like your proposal as well. I think seeing all notifications individually in the profile (with the possible expansion in the header too) would improve the experience significantly. It’s a shame it’s not an easy thing to implement, but hopefully we can see something like this in the future.