shenweichen / DeepCTR

Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
https://deepctr-doc.readthedocs.io/en/latest/index.html
Apache License 2.0
7.57k stars 2.21k forks source link

关于FiBiNet bilinear_type为Field-Interaction Type时输出向量尺寸不同的问题 #103

Closed ae2leowen closed 3 years ago

ae2leowen commented 5 years ago

Describe the question(问题描述) 你好,请教下在FiBiNet网络中,当bilinear_type为Field-Interaction Type时,(如果我没有理解错误的话)输出向量尺寸应该是N F(F-1)/2 T<其中N为batch_size,F为field size, T为embedding size>,这样两边concat后应该就是N F(F-1) T,原论文只有提到concat后是N 2F T,是论文没有详写还是我理解错了,有空帮忙理解下,谢谢

Additional context 另外,我看到作者在layers/interaction.py/class BilinearInteraction(Layer)/compute_output_shape中有写到输出是N F(F-1)/2 T,但看起来好像是对于三种bilinear_type都是这个大小?

Operating environment(运行环境):

shenweichen commented 5 years ago
  1. concat 后应该是 N F(F-1) T
  2. 三种type的输出shape都是一样的