Chiesa had a challenging first season in England after joining Liverpool from Juventus in August last year.
A return to Italy could be beneficial for all parties involved, as Milan can offer Chiesa a spot in the main team without the added pressure of European competition next season.
With the potential arrival of Florian Wirtz from Bayer Leverkusen, Chiesa may face even more limited opportunities to play for Liverpool if he stays.