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