Highlighting parts that are relevant does help me when I do that (but usually I don't want to mess up the book (mess it up in my mind) by doing this.

Something that might help is returning to particularly interesting sections again the same day, or the next day, maybe before bed?