Il mese successivo, la prima linea britannica cominciò a prepararsi per un'attesa offensiva giapponese. Nel marzo, il Giappone invase l'India, e poi assediò Imphal e Kohima. In risposta, la 2a Divisione fu assegnata alla Quattordicesima Armata britannica e gli fu ordinato di spostarsi a Chittagong. Come risultato dei combattimenti a Kohima, la divisione fu deviata verso la provincia di Assam. Dopo una mossa di, la divisione lanciò un contrattacco il 12 aprile. In collaborazione con la 7a Divisione di fanteria indiana, la divisione ha rotto l'assedio giapponese di Kohima
e li costrinse a ritirarsi. Avanzando verso Imphal, la divisione combatté diverse azioni con le forze giapponesi. Il 20 giugno, la divisione si unì alle forze indiane britanniche che erano avanzate da Imphal, ponendo fine all'assedio. Nel corso di questi combattimenti, la divisione aiutò nella distruzione della 31a divisione giapponese e subì 2.125 vittime.