Recently, a 4D Cherns-Simons (CS) theory has been proposed by Costello and Yamazaki as a unified theory of 2D classically integrable field theories. In this talk, I will explain how to derive Yang-Baxter sigma models from this 4D CS theory.