标签聚类损失在遥感影像分类中的应用

苏赋, 于海鹏, 朱威西

Application of label clustering loss in the classification of remote sensing images

SU Fu, YU Haipeng, ZHU Weixi

表1 标签聚类损失训练过程

Tab.1 Label clustering loss training process

算法1:标签聚类损失的正向反向传播算法

1:输入:网络模型最后一层全连接层特征值{

x_{i}

},训练批次大小m,迭代次数T,学习率

μ

和∂_s
2:初始化: softmax损失参数

θ

,类标签中心参数

c_{0}

3: for t=1 to T do
4:计算联合损失:

L_{t o t a l} = L_{s o f t m a x} + λ L_{c}

5:更新softmax损失参数:

θ^{t + 1} = θ^{t} - μ \frac{\partial {L^{t}}_{s o f t m a x}}{\partial θ^{t}}

6:更新类别中心:

{c_{j}}^{t + 1} = {c_{j}}^{t} - \partial_{s} Δ {c_{j}}^{t}

7:更新模型反向传播误差:

\frac{\partial {L^{t}}_{t o t a l}}{\partial {x_{i}}^{t}} = \frac{\partial {L^{t}}_{s o f t m a x}}{\partial {x_{i}}^{t}} + λ \frac{\partial {L^{t}}_{c}}{\partial {x_{i}}^{t}}

8: end for