Using AI for memory improvement