So I went ahead and updated from 502 to 504 a couple of days ago. The shuffle does seem to be working a lot better. It's doing a good job of picking songs through my entire USB library, rather than just in a small subset like before. I haven't noticed any other changes or fixes. Hopefully the good behavior continues.