Krivoshéin fue gravemente herido en la batalla y estuvo recuperándose durante varios meses. Más tarde, en 1944, recibió el mando del 1er Cuerpo mecanizado y luchó en la Operación Bagratión, en el curso de la cual el Grupo de Ejércitos Centro alemán fue prácticamente aniquilado y los alemanes fueron expulsado de Bielorrusia y de la parte oriental de Polonia hasta el río Vístula. Posteriormente combatió en la ofensiva del Vístula-Óder donde en menos de un mes las tropas soviéticos cruzaron la llanura polaca desde el río Vístula hasta alcanzar la Línea Oder-Neisse, situándose a menos de 70 km de Berlín. Los alemanes sufrieron graves bajas tanto en hombres como en material, mientras que el avance soviético dejó aisladas a muchas de sus unidades en Breslau, Poznan, Pomerania y Prusia Oriental.