I've been working on Neural Collaborative Filtering (NCF) recently to build a recommender system. After doing some hyperparameter tuning with various sizes for embedding and dense layers sizes, from 16 all the way to 2048, I realized less complex models tend to work better and usually do a better job at avoiding overfitting. This is not so surprising until now, I suppose.

I know it's generally very contextual and data-dependent. The less complex relationships in the data, the higher chance of overfitting in training the networks with large capacity. But I can't help but wonder, besides halving the size of dense networks, if there is any rule of thumb for the size of such networks. How do you usually go about choosing the size of the network for NCF models?


Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.