AI and digital infrastructure are likely to be key focus areas in the upcoming Union Budget, with major measures expected to boost startups and innovation.