Closed Gangtaro closed 2 years ago
๋๋ค ์ถ์ผ๋ฉด ์๋์ธ์
UltraGCN ๋ชจ๋ธ์ ๋ ผ๋ฌธ์ ํ ๋ฒ ์ฝ์ด๋ดค์ต๋๋ค. ๊ฐ๋ ์ LightGCN์ layer๊ฐ 1~4๊ฐ์ฉ ์๋ ๊ฒ์์, layer๊ฐ ์์ผ์๋ก ์ด๋ ๊ฐ์ผ๋ก ์ ์ ์ ์์ดํ ์ ์๋ฒ ๋ฉ ๋ฒกํฐ๊ฐ ํน์ ๊ฐ์ผ๋ก ์๋ ดํ๊ธฐ ๋๋ฌธ์ ๋ง์ Layer์ ๊ณ์ฐ์ด ํ์์์ด์, ์๋ ดํ ์ํ์ layerํ๋๋ง ํ์ํ๋ค๋ ๊ฐ๋ ์์ ์์ํฉ๋๋ค. ํ์ง๋ง ํด๋น ์์๊ณผ ๊ด๋ จ๋ ๋ด์ฉ ์์ฒด๊ฐ Loss ๊ฐ์ ์ด๋ป๊ฒ ํ์ฑํ๋๋์ ์ข์ฐ ๋๋ฏ๋ก, ๊ธฐ์กด์ DKT์ Binary Classification ๋ฐฉ์์ ์ ์ฉํ๊ธฐ ํ๋ค๋ค๊ณ ์๊ฐํฉ๋๋ค.
ํด๋น ๋ด์ฉ์ #9 RecBole lightgcn ์ฑ๋ฅ์ด Baseline์ lightgcn ์ ์ฑ๋ฅ๊ณผ ์ฐจ์ด๋๋ ์ด์ ๋ฅผ ์ดํดํ๋ฉด ๋ฉ๋๋ค. RecBole์ ๊ฒฝ์ฐ, ํ์๋ ๋ฌธ์ ๋ง interaction์ผ๋ก ๊ฐ์ ํ๊ณ ํด๋น ๋งํฌ์ ์ฐ๊ฒฐ์ฑ์ ๊ฐ์ง๊ณ ๋ง ํ์ต์ ์งํํ์ต๋๋ค. ํ์ง๋ง, Baseline ์ ๊ฒฝ์ฐ์๋ Negative sampling์ ํด์ฃผ์ง ์๊ณ , ์์ดํ ๊ณผ ์ ์ ์๋ฒ ๋ฉ ๋ฒกํฐ์ dot product์ ๊ฐ์ label (0, 1)๊ณผ BCE loss ๊ฐ์ objective๋ก ํ์ฌ ํ์ตํ๋ ๋ฐฉ๋ฒ์ ์ฑํํ์ฌ, ๊ธฐ์กด์ lightgcn์ ํ์ต ๋ฐฉ๋ฒ๊ณผ๋ ๋ค๋ฅธ ๋ฐฉ๋ฒ์ ์ฌ์ฉํ๋ ๊ฒ์ ๋๋ค. ์ด๊ฒ ๋ ์ฑ๋ฅ์ด ์ข์์ต๋๋ค.
์ด๊ฒ์ ๋ค์ ๋งํ์๋ฉด, ํด๋น ๋ํ์ ๋ฌธ์ ์ํฉ์ ๋ ์ฌ๋ฐ๋ฅธ ๋ฐฉ๋ฒ์ผ๋ก ๋ชจ๋ธ์ Loss ๊ฐ์ ๊ตฌํ๊ธฐ ์ํด์๋ BCE Loss ๊ฐ์ ์ฑํํด์ผํฉ๋๋ค. ํ์ง๋ง, UltraGCN์ ๊ฒฝ์ฐ, ์ ์ด์ ๊ฐ๋ ์์ฒด๊ฐ LightGCN์ Recommeder System ์ํฉ์ ์์ ํ๊ณ ์ด์ ๋ํ ๊ทนํ๊ฐ์ ์ถ์ ํ์ฌ ๊ทธ๊ฒ์ Loss ๊ณ์ฐ์ ์ ์ฉํ๋ ๊ฒ์ ๋๋ค.
๋ฐ๋ผ์ ๋ ผ๋ฌธ๊ณผ ์์ ์ฝ๋์์ ์ฃผ์ด์ง Loss ๊ฐ ๊ณ์ฐ์ ๋ฌด์ํ๊ฒ ๋๋ฉด ์๋ง์ ๋ด์ฉ์ ์์ด๋ฒ๋ฆฌ๊ฒ ๋๊ณ ๋จ์ GCN Classification model๊ณผ ๋ค๋ฆ์ด ์์ด์ง๋ค๊ณ ์๊ฐํฉ๋๋ค.
๊ทธ๋์ ์์ ๊ฐ์ ์ด์ ์ ๋ฐ๋ฅธ ์ฑ๋ฅ์ ๋ณด์ฅํ ์ ์์ ๊ฒ ๊ฐ๋ค๋ ๋ฆฌ์คํฌ์ ์๊ฐ์ ์ธ ์์ ๋๋ฌธ์ UltraGCN ๋ชจ๋ธ ๊ฐ๋ฐ์ ์ ์ ์ค๋จํ๋๋ก ํ๊ฒ ์ต๋๋ค.
๊ณ ์ ๋ง์ผ์ จ์ต๋๋ค!
๐ก Issue
lightgcn baseline์ Ultragcn ๋ชจ๋ธ์ ์ถ๊ฐํ๊ณ ์ ํฉ๋๋ค. ์ค๋ ํผ์ด์ธ์ ๊น์ง ์์์ ์์์, ์ ๊ฐ ํ๋๋ก ํ๊ฒ ์ต๋๋ค.
*์ถ๊ฐ
๐ TODO
๐ Reference