當(dāng)前位置:首頁 > 學(xué)習(xí)資源 > 講師博文 > 信息增益
一、信息增益的定義
信息增益(Information Gain)是決策樹算法中用于特征選擇的一種重要度量標(biāo)準(zhǔn)。它衡量了在使用某個特征進(jìn)行數(shù)據(jù)集劃分之后,數(shù)據(jù)集不確定性(或熵)的減少程度。
簡單來說,信息增益表示了由于特征的存在而使得數(shù)據(jù)集分類變得更加明確的信息量。
具體來說,信息增益的定義為:
信息增益=父節(jié)點(diǎn)的信息熵−子節(jié)點(diǎn)的加權(quán)平均信息熵
其中:
l 父節(jié)點(diǎn)的信息熵是指在使用特征進(jìn)行劃分之前,數(shù)據(jù)集整體的不確定性或混亂程度,通常使用香農(nóng)熵(Shannon Entropy)來衡量。
l 子節(jié)點(diǎn)的加權(quán)平均信息熵是指在使用特征進(jìn)行劃分之后,各個子數(shù)據(jù)集(即劃分后的各個分支)的信息熵的加權(quán)平均值,權(quán)重通常根據(jù)子數(shù)據(jù)集的大小來確定。
信息增益越大,說明使用該特征進(jìn)行劃分后,數(shù)據(jù)集的不確定性減少得越多,即該特征對分類的貢獻(xiàn)越大。因此,在構(gòu)建決策樹時,通常會選擇信息增益最大的特征作為當(dāng)前節(jié)點(diǎn)的分裂特征,以最大化地減少數(shù)據(jù)集的不確定性,從而提高分類的準(zhǔn)確性。
二、信息增益的計(jì)算步驟
1. 計(jì)算父節(jié)點(diǎn)的信息熵:
使用訓(xùn)練集中所有樣本的類別信息計(jì)算父節(jié)點(diǎn)的信息熵。信息熵是用于度量一個系統(tǒng)的不確定性或無序程度的概念。計(jì)算公式為:
其中,H(D)是數(shù)據(jù)集 DD 的熵,pi是數(shù)據(jù)集 D中第 i類的概率,k 是類別的總數(shù)。
2. 計(jì)算特征的條件熵:
對于每個可能的分裂特征,計(jì)算該特征條件下的信息熵。這涉及將訓(xùn)練集中的樣本劃分成不同的子集,并計(jì)算每個子集的信息熵。然后,根據(jù)每個子集在數(shù)據(jù)集中的比例,計(jì)算加權(quán)平均的條件熵。計(jì)算公式為:
其中,H(D∣A)是在特征 A 的條件下的條件熵,Values(A)是特征 A 的所有取值,DvDv是 在特征 A的取值 v上的數(shù)據(jù)子集,∣Dv∣和 ∣D∣分別是數(shù)據(jù)子集 Dv和數(shù)據(jù)集 D的大 小。
3. 計(jì)算信息增益:
信息增益是父節(jié)點(diǎn)信息熵減去由于特征分裂而導(dǎo)致的子節(jié)點(diǎn)的加權(quán)平均信息熵。計(jì)算公式為:
其中,IG(D,A) 是在特征 A上的信息增益。
三、信息增益在構(gòu)建決策樹時的重要性
(一)特征選擇:
在構(gòu)建決策樹時,信息增益用于選擇最優(yōu)特征來劃分?jǐn)?shù)據(jù)集。
對于每個節(jié)點(diǎn),計(jì)算所有候選特征的信息增益,并選擇信息增益最大的特征作為該節(jié)點(diǎn)的分裂特征。這可以確保每個分裂都能最大程度地減少數(shù)據(jù)集的不確定性,從而提高分類的準(zhǔn)確性。
(二)避免過擬合:
信息增益不僅考慮了特征對數(shù)據(jù)集分類能力的提升,還通過信
息熵和條件熵的計(jì)算,隱含地考慮了數(shù)據(jù)的純度。因此,使用信息增益作為特征選擇的標(biāo)準(zhǔn),有助于避免選擇過于復(fù)雜的特征,從而減少過擬合的風(fēng)險(xiǎn)。
(三)構(gòu)建高效的決策樹:
通過信息增益進(jìn)行特征選擇,可以逐層分裂數(shù)據(jù)集,使得每個
葉節(jié)點(diǎn)盡可能純凈。這有助于構(gòu)建更加簡潔和高效的決策樹模型,提高分類和預(yù)測的速度和準(zhǔn)確性。
總之,信息增益在決策樹算法中扮演著至關(guān)重要的角色,它通過量化特征對數(shù)據(jù)集分類能力提升的貢獻(xiàn)程度,指導(dǎo)特征選擇過程,從而構(gòu)建出高效且準(zhǔn)確的決策樹模型。
四、信息增益的計(jì)算示例
假設(shè)我們有一個關(guān)于是否出去玩的決策樹數(shù)據(jù)集,其中包含以下屬性:Outlook(天氣狀況)、Temperature(溫度)、Humidity(濕度)和Windy(是否刮風(fēng)),目標(biāo)變量是Play(是否出去玩)。
首先,我們計(jì)算整個數(shù)據(jù)集的信息熵(原始狀態(tài)的信息量):數(shù)據(jù)集包含14個實(shí)例,其中9個正例(yes)和5個負(fù)例(no)。
接下來,我們使用Outlook屬性來劃分?jǐn)?shù)據(jù)集,并計(jì)算劃分后的信息熵。Outlook屬性有三個取值:sunny、overcast和rain。
1. 對于Outlook=sunny,有5個實(shí)例,其中2個正例和3個負(fù)例:
2. 對于Outlook=overcast,有4個實(shí)例,全部是正例:
3. 對于Outlook=rain,有5個實(shí)例,其中3個正例和2個負(fù)例:
然后,我們計(jì)算劃分后的信息總量,即加權(quán)平均信息熵:
最后,我們計(jì)算信息增益:
類似地,我們可以計(jì)算其他屬性的信息增益,并選擇信息增益最大的屬性作為當(dāng)前節(jié)點(diǎn)的分裂屬性。
請注意,上述計(jì)算中的數(shù)值可能因四舍五入而有輕微差異,但整體思路和步驟是正確的。在實(shí)際應(yīng)用中,應(yīng)使用精確的計(jì)算結(jié)果來構(gòu)建決策樹。