The models you ate looking for is the *V models:
ATmega644PV, ATmega1284P, ATmega640V, ATmega641V, ATmega1280V, ATmega1281V, ATmega2560V, ATmega2561V.
The best option is to maybe choose a microcontroller in a DIP package for easy soldering. This leaves out all except for the ATmega644PV and ATmega1284P. These are both great microcontrollers with great Arduino support, thanks to (my very own) MightyCore.
If the ATmega1284 isn't a whole lot more expensive, you should definitely go for this one. It has 16 kB RAM, which is 8 (!!) times more than an ATmega328!