USGS-R / river-dl

Deep learning model for predicting environmental variables on river systems
Creative Commons Zero v1.0 Universal
21 stars 15 forks source link

Implement adapting auxiliary loss based on gradient similarity #80

Closed jsadler2 closed 3 years ago

jsadler2 commented 3 years ago

Implement technique for adapting the auxiliary loss based on the similarity between the main gradient and the auxiliary gradient. The method is described in this paper by Du et. al. I actually like the slight modification described in this [blog post(https://vivien000.github.io/blog/journal/learning-though-auxiliary_tasks.html) by Tran-Thien, so I will be implementing this version.