AC Milan take on Atletico at the Vicente Calderon on Tuesday night hoping to overturn a 1-0 deficit from the first-leg at the San Siro.
And, according to French newspaper Le 10 Sport, Balotelli wants a goal-scoring return to form for the Rossoneri to help convince the club's officials to ignore potential bids from suitors.
Chelsea have been heavily linked with the Italian forward, with his former-Inter Milan boss Jose Mourinho thought to be plotting a reunion at Stamford Bridge.
The two worked together in Italy, before Balotelli departed for Manchester City in 2010. He spent three years at the club, winning a Premier League title, but left for AC Milan in January 2013.
Despite an impressive start to his career back in Italy, AC Milan's struggles this season have impacted upon Balotelli.
He has 10 goals in 19 league appearances so far this season, and leads the club's goalscoring charts.
But he is reportedly not living up to the expectations of Silvio Berlusconi and is only now returning from injury - Balotelli picked up a shoulder complaint against Atletico Madrid last month.
That prompted fresh speculation linking him with Chelsea but Balotelli is thought to favour an extended stay in Italy.