reviseBased on this idea a multi-task learning MTL module is proposed and embedded in each stage of encoder for jointly learning to detect edges of rail surface defects and segment the defect regions

In order to simultaneously detect edges of rail surface defects and segment the defect regions, we propose the integration of a multi-task learning (MTL) module within each stage of the encoder.