Une unité disciplinaire ou unité pénitentiaire est une formation militaire dont les soldats sont des condamnés purgeant en son sein la peine prononcée par une juridiction militaire. Selon sa taille, une telle unité peut être un bataillon disciplinaire, une compagnie disciplinaire ou un régiment disciplinaire.