The game saves your progress a few seconds after the end of each wave.
For keys spawning at the end of the wave, it’s possible your progress is saved before you collect all of them (i.e., some of the keys are collected during the wave that just ended, and some during the upcoming wave). If you use “End Session” then you will lose the “upcoming wave” keys.