Storing　multimode　transverse　images　in　atomic　media　is　crucial　in　constructing　large-scale　quantum　networks.　A　major　obstacle　of　storing　transverse　images　in　the　thermal　atomic　vapor　is　the　distortion　of　the　retrieved　images　caused　by　atomic　diffusion.　Here,　we　demonstrate　the　combination　of　genetic　algorithm　with　the　phase-shift　lithography　method　to　construct　the　optimal　phase　for　an　arbitrary　transverse　image　that　can　diminish　the　effect　of　diffusion.　Theoretical　simulations　and　experimental　results　manifest　that　the　retrieved　images＇　resolution　can　be　substantially　improved　when　carrying　the　optimal　phases.　Our　scheme　is　efficient　and　straightforward　and　can　be　extensively　applied　in　storing　complex　transverse　multimodes　in　diffusion　media.