The chancellor will unveil AI spending aimed at shoring up the UK's defences in Wednesday's Budget.