金融界2024年2月9日消息,據國家知識產權局公告,清華大學申請一項名為“基於注意力機制的小樣本圖像分類方法及裝置“,公開號CN117542075A,申請日期為2023年10月。
專利摘要顯示,本發明涉及小樣本圖像識別技術領域,特別涉及一種基於注意力機制的小樣本圖像分類方法及裝置,其中,方法包括:獲取並增強不同種類的待分類原始圖像數據,將增強圖像數據劃分為支持集、查詢集樣本;利用預先構建的目標顯著性模型,提取支持集樣本的目標主體顯著圖像;將目標主體顯著圖像和支持集樣本進行特征提取,得到不同網絡下的圖像特征,並進行融合得到分類超平面;根據分類超平面更新支持集樣本的原型,並計算查詢集樣本與支持集樣本的原型的距離,以將查詢集樣本分類到支持集樣本的類別中。由此,解決了小樣本圖像識別的數據量少,難以訓練深層的神經網絡,造成小樣本深度學習模型的泛化能力低等問題。