Pile-up mitigation using attention. Issue 2 (1st June 2022)