一種深度學習模型的訓練方法及裝置

基本信息

申請?zhí)?/td> CN201810258256.0 申請日 -
公開(公告)號 CN108734193A 公開(公告)日 2018-11-02
申請公布號 CN108734193A 申請公布日 2018-11-02
分類號 G06K9/62;G06N99/00 分類 計算;推算;計數;
發(fā)明人 李誠;周曉;朱才志 申請(專利權)人 合肥麟圖信息科技有限公司
代理機構 北京柏杉松知識產權代理事務所(普通合伙) 代理人 馬敬;項京
地址 230000 安徽省合肥市高新區(qū)望江西路800號創(chuàng)新產業(yè)園A3樓305-2號
法律狀態(tài) -

摘要

摘要 本發(fā)明實施例提供了一種深度學習模型的訓練方法及裝置,方法為:首先訓練得到中間深度學習模型;計算訓練樣本集中各個樣本的特征向量以及計算中心點距離目標函數的中間參數的初始值;將中心點距離目標函數添加到中間深度學習模型中并加載中間參數的初始值,得到目標深度學習模型;利用當前目標深度學習模型,計算當前批次數據中各個樣本的特征向量,并更新中間參數的參數值;計算中心點距離目標函數的函數值和分類目標函數的函數值,并判斷是否符合結束訓練的條件;如果否,調整當前目標深度學習模型的參數,導入下一批次數據并進行模型訓練;如果是,結束訓練。應用本發(fā)明實施例提供的方案可以提升深度學習模型的分類準確率。