El mes següent, la línia de front britànica va començar a preparar-se per a una ofensiva japonesa. Al març, el Japó va envair l'Índia, i després va assetjar Imphal i Kohima. En resposta, la 2a Divisió va ser assignada al 14è Exèrcit britànic i va rebre l'ordre de traslladar-se a Chittagong. Com a resultat dels combats a Kohima, la divisió va ser desviada a la província d'Assam. Després d'un moviment de, la divisió va llançar un contraatac el 12 d'abril. Juntament amb la 7a Divisió d'Infanteria Índia, la divisió va trencar el setge japonès de Kohima.
i els va forçar a retirar-se. Avançant cap a Imphal, la divisió va lluitar en diverses accions amb les forces japoneses. El 20 de juny, la divisió es va unir amb les forces indies britàniques que havien avançat des d'Imphal, el que va acabar amb el setge. En el transcurs d'aquest combat, la divisió va ajudar a la destrucció de la 31a divisió japonesa, i va patir 2.125 baixes.