Villa have conceded more goals from set pieces than any other team in the top flight and fixing the problem will be key to hopes of escaping the drop.
Grealish said: “The past five or six weeks we have been having Zoom calls twice a week. That has been about everything, the tactical side of things but also the mental side of things.
“We have been working on things in training we haven’t been so good at this season. For example set pieces, we have not been good attacking or defending them so we have been working on things we need to improve on.”
Villa, who sit 19th in the table, will get the ball rolling on the Premier League’s restart when they host Sheffield United on June 17.
Victory would lift them out of the bottom three and Grealish knows the onus is on him to perform.
He added: “I don’t want to say this in an arrogant way, but I embrace (the pressure). I love the fact it’s like that, if I’m honest. Being the local lad and captain of the club, I absolutely love it and try to take it in my stride which I think I have done this season.
“I probably didn’t start the season as well as I wanted to. It took me time to get up to the speed of things and it took a lot of our players the same. We had a lot of players coming from abroad.
“We are going straight into a relegation battle but we are all focused on that and we are all ready.”