35 KiB
Neural Collaborative Filtering (NCF) for PyTorch
This repository provides a script and recipe to train the Neural Collaborative Filtering (NCF) model to achieve state of the art accuracy, and is tested and maintained by NVIDIA.
Table of Contents
- Model overview
- Setup
- Quick Start Guide
- Advanced
- Performance
- Release notes
Model overview
The NCF model focuses on providing recommendations, also known as collaborative filtering with implicit feedback. The training data for this model should contain binary information about whether a user interacted with a specific item. NCF was first described by Xiangnan He, Lizi Liao, Hanwang Zhang, Liqiang Nie, Xia Hu and Tat-Seng Chua in the Neural Collaborative Filtering paper.
The implementation in this repository focuses on the NeuMF instantiation of the NCF architecture. We modified it to use dropout in the FullyConnected layers. This reduces overfitting and increases the final accuracy. Training the other two instantiations of NCF (GMF and MLP) is not supported.
Contrary to the original paper, we benchmark the model on the larger ML-20m dataset instead of using the smaller ML-1m dataset as we think this is more realistic of production type environments. However, using the ML-1m dataset is also supported.
This model is trained with mixed precision using Tensor Cores on Volta, Turing, and the NVIDIA Ampere GPU architectures. Therefore, researchers can get results 2x faster than training without Tensor Cores, while experiencing the benefits of mixed precision training.
Model architecture
This model is based mainly on Embedding and FullyConnected layers. The control flow is divided into two branches:
- Multi Layer Perceptron (MLP) branch, which transforms the input through FullyConnected layers with ReLU activations and dropout.
- Matrix Factorization (MF) branch, which performs collaborative filtering factorization. Each user and each item has two embedding vectors associated with it -- one for the MLP branch and the other for the MF branch.
The outputs from those branches are concatenated and fed to the final FullyConnected layer with sigmoid activation. This can be interpreted as a probability of a user interacting with a given item.
Figure 1. The architecture of a Neural Collaborative Filtering model. Taken from the Neural Collaborative Filtering paper.
Default configuration
The following features were implemented in this model:
- Automatic Mixed Precision (AMP)
- Data-parallel multi-GPU training and evaluation
- Dropout
- Gradient accumulation
The following performance optimizations were implemented in this model:
- FusedAdam optimizer
- Approximate train negative sampling
- Caching all the positive training samples in the device memory
Feature support matrix
The following features are supported by this model:
Feature | NCF PyTorch |
---|---|
Automatic Mixed Precision (AMP) | Yes |
Multi-GPU training with Distributed Data Parallel (DDP) | Yes |
Fused Adam | Yes |
Features
- Automatic Mixed Precision - This implementation of NCF uses AMP to implement mixed precision training. It allows us to use FP16 training with FP32 master weights by modifying just 3 lines of code.
- Multi-GPU training with Distributed Data Parallel - uses Apex's DDP to implement efficient multi-GPU training with NCCL.
- Fused Adam - We use a special implementation of the Adam implementation provided by the Apex package. It fuses some operations for faster weight updates. Since NCF is a relatively lightweight model with a large number of parameters, we’ve observed significant performance improvements from using FusedAdam.
Mixed precision training
Mixed precision is the combined use of different numerical precisions in a computational method. Mixed precision training offers significant computational speedup by performing operations in half-precision format, while storing minimal information in single-precision to retain as much information as possible in critical parts of the network. Since the introduction of tensor cores in Volta, and following with both the Turing and Ampere architectures, significant training speedups are experienced by switching to mixed precision -- up to 3x overall speedup on the most arithmetically intense model architectures. Using mixed precision training requires two steps:
- Porting the model to use the FP16 data type where appropriate.
- Adding loss scaling to preserve small gradient values.
The ability to train deep learning networks with lower precision was introduced in the Pascal architecture and first supported in CUDA 8 in the NVIDIA Deep Learning SDK.
For information about:
- How to train using mixed precision, see the Mixed Precision Training paper and Training With Mixed Precision documentation.
- Techniques used for mixed precision training, see the Mixed-Precision Training of Deep Neural Networks blog.
- How to access and enable AMP for TensorFlow, see Using TF-AMP from the TensorFlow User Guide.
- APEX tools for mixed precision training, see the NVIDIA Apex: Tools for Easy Mixed-Precision Training in PyTorch.
Enabling mixed precision
Using the Automatic Mixed Precision (AMP) package requires two modifications in the source code.
The first one is to initialize the model and the optimizer using the amp.initialize
function:
model, optimizer = amp.initialize(model, optimizer, opt_level="O2"
keep_batchnorm_fp32=False, loss_scale='dynamic')
The second one is to use the AMP's loss scaling context manager:
with amp.scale_loss(loss, optimizer) as scaled_loss:
scaled_loss.backward()
Enabling TF32
TensorFloat-32 (TF32) is the new math mode in NVIDIA A100 GPUs for handling the matrix math also called tensor operations. TF32 running on Tensor Cores in A100 GPUs can provide up to 10x speedups compared to single-precision floating-point math (FP32) on Volta GPUs.
TF32 Tensor Cores can speed up networks using FP32, typically with no loss of accuracy. It is more robust than FP16 for models which require high dynamic range for weights or activations.
For more information, refer to the TensorFloat-32 in the A100 GPU Accelerates AI Training, HPC up to 20x blog post.
TF32 is supported in the NVIDIA Ampere GPU architecture and is enabled by default.
Setup
The following section lists the requirements in order to start training the Neural Collaborative Filtering model.
Requirements
This repository contains Dockerfile which extends the PyTorch NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following components:
- NVIDIA Docker
- PyTorch 20.06-py3+ NGC container
- Supported GPUs:
For more information about how to get started with NGC containers, see the following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning Documentation:
Running PyTorch
For those unable to use the [framework name] NGC container, to set up the required environment or create your own container, see the versioned NVIDIA Container Support Matrix.
Quick Start Guide
To train your model using mixed or TF32 precision with Tensor Cores or using FP32, perform the following steps using the default parameters of the NCF model on the ML-20m dataset. For the specifics concerning training and inference, see the Advanced section.
- Clone the repository.
git clone https://github.com/NVIDIA/DeepLearningExamples
cd DeepLearningExamples/PyTorch/Recommendation/NCF/
- Build an NCF PyTorch Docker container.
After Docker is setup, you can build the NCF image with:
docker build . -t nvidia_ncf
- Start an interactive session in the NGC container to run preprocessing/training and inference.
The NCF PyTorch container can be launched with:
mkdir data
docker run --runtime=nvidia -it --rm --ipc=host -v ${PWD}/data:/data nvidia_ncf bash
This will launch the container and mount the ./data
directory as a volume to the /data
directory inside the container.
Any datasets and experiment results (logs, checkpoints etc.) saved to /data
will be accessible
in the ./data
directory on the host.
- Download and preprocess the data.
Preprocessing consists of downloading the data, filtering out users that have less than 20 ratings (by default), sorting the data and dropping the duplicates. The preprocessed train and test data is then saved in PyTorch binary format to be loaded just before training.
Note: Preprocessing requires PyTorch and should therefore be run inside the Docker container.
No data augmentation techniques are used.
To download and preprocess the ML-20m dataset you can run:
./prepare_dataset.sh
Note: This command will return immediately without downloading anything if the data is already present in the /data
directory.
This will store the preprocessed training and evaluation data in the ./data
directory so that it can be later
used to train the model (by passing the appropriate --data
argument to the ncf.py
script).
- Start training.
After the Docker container is launched, the training with the default hyperparameters can be started with:
python -m torch.distributed.launch --nproc_per_node=8 --use_env ncf.py --data /data/cache/ml-20m --checkpoint_dir /data/checkpoints/
This will result in a checkpoint file being written to /data/checkpoints/model.pth
.
- Start validation/evaluation.
The trained model can be evaluated by passing the --mode
test flag to the run.sh
script:
python -m torch.distributed.launch --nproc_per_node=1 --use_env ncf.py --data /data/cache/ml-20m --mode test --load_checkpoint_path /data/checkpoints/model.pth
Advanced
The following sections provide greater details of the dataset, running training and inference, and the training results.
Scripts and sample code
The ncf.py
script contains most of the training and validation logic. Data loading and preprocessing code is located in dataloading.py
.
The model architecture is defined in neumf.py
. Some initial data preprocessing is located in convert.py
.
The logger directory contains simple bookkeeping utilities for storing training results.
Command-line options
To see the full list of available options and their descriptions, use the -h
or --help
command line option, for example:
python ncf.py --help
The following example output is printed when running the sample:
usage: ncf.py [-h] [--data DATA] [-e EPOCHS] [-b BATCH_SIZE]
[--valid_batch_size VALID_BATCH_SIZE] [-f FACTORS]
[--layers LAYERS [LAYERS ...]] [-n NEGATIVE_SAMPLES]
[-l LEARNING_RATE] [-k TOPK] [--seed SEED]
[--threshold THRESHOLD]
[--beta1 BETA1] [--beta2 BETA2] [--eps EPS] [--dropout DROPOUT]
[--checkpoint_dir CHECKPOINT_DIR] [--mode {train,test}]
[--grads_accumulated GRADS_ACCUMULATED] [--amp]
[--local_rank LOCAL_RANK]
Train a Neural Collaborative Filtering model:
optional arguments:
-h, --help show this help message and exit
--data DATA Path to test and training data files
-e EPOCHS, --epochs EPOCHS
Number of epochs for training
-b BATCH_SIZE, --batch_size BATCH_SIZE
Number of examples for each iteration
--valid_batch_size VALID_BATCH_SIZE
Number of examples in each validation chunk
-f FACTORS, --factors FACTORS
Number of predictive factors
--layers LAYERS [LAYERS ...]
Sizes of hidden layers for MLP
-n NEGATIVE_SAMPLES, --negative_samples NEGATIVE_SAMPLES
Number of negative examples per interaction
-l LEARNING_RATE, --learning_rate LEARNING_RATE
Learning rate for optimizer
-k TOPK, --topk TOPK Rank for test examples to be considered a hit
--seed SEED, -s SEED Manually set random seed for torch
--threshold THRESHOLD, -t THRESHOLD
Stop training early at threshold
--beta1 BETA1, -b1 BETA1
Beta1 for Adam
--beta2 BETA2, -b2 BETA2
Beta1 for Adam
--eps EPS Epsilon for Adam
--dropout DROPOUT Dropout probability, if equal to 0 will not use
dropout at all
--checkpoint_dir CHECKPOINT_DIR
Path to the directory storing the checkpoint file
--mode {train,test} Passing "test" will only run a single evaluation,
otherwise full training will be performed
--grads_accumulated GRADS_ACCUMULATED
Number of gradients to accumulate before performing an
optimization step
--amp Enable mixed precision training
--local_rank LOCAL_RANK
Necessary for multi-GPU training
Getting the data
The NCF model was trained on the ML-20m dataset. For each user, the interaction with the latest timestamp was included in the test set and the rest of the examples are used as the training data.
This repository contains the ./prepare_dataset.sh
script which will automatically download and preprocess the training and validation datasets.
By default, data will be downloaded to the /data
directory. The preprocessed data will be placed in /data/cache
.
Dataset guidelines
The required format of the data is a CSV file with three columns: user_id
, item_id
and timestamp
. This CSV should contain only the positive examples, in other words,
the ones for which an interaction between a user and an item occurred. The negatives will be sampled during the training and validation.
Multi-dataset
This implementation is tuned for the ML-20m and ML-1m datasets. Using other datasets might require tuning some hyperparameters (for example, learning rate, beta1 and beta2).
If you'd like to use your custom dataset you can do it by adding support for it in the prepare_dataset.sh
and download_dataset.sh
scripts.
The performance of the model depends on the dataset size. Generally, the model should scale better for datasets containing more data points. For a smaller dataset you might experience slower performance.
ML-1m
To download, preprocess and train on the ML-1m dataset run:
./prepare_dataset.sh ml-1m
python -m torch.distributed.launch --nproc_per_node=8 --use_env ncf.py --data /data/cache/ml-1m
Training process
The name of the training script is ncf.py
. Because of the multi-GPU support, it should always be run with the torch distributed launcher like this:
python -m torch.distributed.launch --nproc_per_node=<number_of_gpus> --use_env ncf.py --data <path_to_dataset> [other_parameters]
The main result of the training are checkpoints stored by default in /data/checkpoints/
. This location can be controlled
by the --checkpoint_dir
command-line argument.
The validation metric is Hit Rate at 10 (HR@10) with 100 test negative samples. This means that for each positive sample in the test set 100 negatives are sampled. All resulting 101 samples are then scored by the model. If the true positive sample is among the 10 samples with highest scores we have a "hit" and the metric is equal to 1, otherwise it's equal to 0. The HR@10 metric is the number of hits in the entire test set divided by the number of samples in the test set.
Inference process
Inference can be launched with the same script used for training by passing the --mode test
flag:
python -m torch.distributed.launch --nproc_per_node=<number_of_gpus> --use_env ncf.py --data <path_to_dataset> --mode test [other_parameters]
The script will then:
- Load the checkpoint from the directory specified by the
--checkpoint_dir
directory - Run inference on the test dataset
- Compute and print the validation metric
Performance
The performance measurements in this document were conducted at the time of publication and may not reflect the performance achieved from NVIDIA’s latest software release. For the most up-to-date performance measurements, go to NVIDIA Data Center Deep Learning Product Performance.
Benchmarking
Training performance benchmark
NCF training on NVIDIA DGX systems is very fast, therefore, in order to measure train and validation throughput, you can simply run the full training job with:
./prepare_dataset.sh
python -m torch.distributed.launch --nproc_per_node=8 --use_env ncf.py --data /data/cache/ml-20m --epochs 5
At the end of the script, a line reporting the best train throughput is printed.
Inference performance benchmark
Validation throughput can be measured by running the full training job with:
./prepare_dataset.sh
python -m torch.distributed.launch --nproc_per_node=8 --use_env ncf.py --data /data/cache/ml-20m --epochs 5
The best validation throughput is reported to the standard output.
Results
The following sections provide details on how we achieved our performance and accuracy in training and inference.
Training accuracy results
Training accuracy: NVIDIA DGX A100(8x A100 40GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX A100 (8x A100 40GB) GPUs.
GPUs | Batch size / GPU | Accuracy - TF32 | Accuracy - mixed precision | Time to train - TF32 | Time to train - mixed precision | Time to train speedup (TF32 to mixed precision) |
---|---|---|---|---|---|---|
1 | 1048576 | 0.958805 | 0.958744 | 141.2 | 96.0 | 1.47 |
8 | 131072 | 0.958916 | 0.959045 | 31.99 | 25.21 | 1.27 |
Training accuracy: NVIDIA DGX-1 (8x V100 16GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX-1 with 8x V100 16GB GPUs.
The following table lists the best hit rate at 10 for DGX-1 with 8 V100 16GB GPUs. It also shows the average time to reach this HR@10 across 5 random seeds. The training time was measured excluding data downloading, preprocessing, validation data generation and library initialization times.
GPUs | Batch size / GPU | Accuracy - FP32 | Accuracy - mixed precision | Time to train - FP32 (s) | Time to train - mixed precision (s) | Time to train speedup (FP32 to mixed precision) |
---|---|---|---|---|---|---|
1 | 1,048,576 | 0.958944 | 0.959093 | 304.922339 | 145.285286 | 2.10 |
8 | 131,072 | 0.959029 | 0.959107 | 55.120696 | 35.165247 | 1.58 |
To reproduce this result, start the NCF Docker container interactively and run:
./prepare_dataset.sh
python -m torch.distributed.launch --nproc_per_node=8 --use_env ncf.py --data /data/cache/ml-20m
Training accuracy: NVIDIA DGX-1 (8x V100 32GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX-1 with 8x V100 32GB GPUs.
The following table lists the best hit rate at 10 for DGX-1 with 8 V100 16GB GPUs. It also shows the average time to reach this HR@10 across 5 random seeds. The training time was measured excluding data downloading, preprocessing, validation data generation and library initialization times.
GPUs | Batch size / GPU | Accuracy - FP32 | Accuracy - mixed precision | Time to train - FP32 (s) | Time to train - mixed precision (s) | Time to train speedup (FP32 to mixed precision) |
---|---|---|---|---|---|---|
1 | 1,048,576 | 0.959172 | 0.958409 | 309.942810 | 153.940982 | 2.01 |
8 | 131,072 | 0.958905 | 0.959228 | 56.922492 | 38.121914 | 1.49 |
Here's an example validation accuracy curve for mixed precision vs single precision on DGX-1 with 8 V100 32GB GPUs:
To reproduce this result, start the NCF Docker container interactively and run:
./prepare_dataset.sh
python -m torch.distributed.launch --nproc_per_node=8 --use_env ncf.py --data /data/cache/ml-20m
Training accuracy: NVIDIA DGX-2 (16x V100 32GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX-1 with 8x V100 16GB GPUs.
The following table lists the best hit rate at 10 for DGX-1 with 8 V100 16GB GPUs. It also shows the average time to reach this HR@10 across 5 random seeds. The training time was measured excluding data downloading, preprocessing, validation data generation and library initialization times.
GPUs | Batch size / GPU | Accuracy - FP32 | Accuracy - mixed precision | Time to train - FP32 (s) | Time to train - mixed precision (s) | Time to train speedup (FP32 to mixed precision) |
---|---|---|---|---|---|---|
1 | 1,048,576 | 0.958794 | 0.958873 | 290.648888 | 143.328581 | 2.03 |
8 | 131,072 | 0.959152 | 0.959150 | 53.792118 | 34.838838 | 1.54 |
16 | 65,536 | 0.959248 | 0.958812 | 40.925270 | 30.214568 | 1.35 |
To reproduce this result, start the NCF Docker container interactively and run:
./prepare_dataset.sh
python -m torch.distributed.launch --nproc_per_node=16 --use_env ncf.py --data /data/cache/ml-20m
Training stability test
The histogram below shows the best HR@10 achieved for 400 experiments using mixed precision and 400 experiments using single precision. Mean HR@10 for mixed precision was equal to 0.95868 and for single precision it was equal to 0.95867.
Training performance results
Training performance: NVIDIA DGX A100 (8x A100 40GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX A100 (8x A100 40GB) GPUs. Performance numbers (in items per second) were averaged over an entire training epoch.
GPUs | Batch size / GPU | Throughput - TF32 | Throughput - mixed precision | Throughput speedup (TF32 - mixed precision) | Strong scaling - TF32 | Strong scaling - mixed precision |
---|---|---|---|---|---|---|
1 | 1,048,576 | 22.64M | 33.59M | 1.484 | 1 | 1 |
8 | 131,072 | 105.8M | 137.8M | 1.302 | 4.67 | 4.10 |
Training performance: NVIDIA DGX-1 (8x V100 16GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX-1 with 8x V100 16GB GPUs.
The following table shows the best training throughput:
GPUs | Batch Size / GPU | Throughput - FP32 (samples / s) | Throughput - Mixed precision (samples /s) | Throughput Speedup (FP32 to Mixed precision) | Strong Scaling - FP32 | Strong scaling - Mixed precision |
---|---|---|---|---|---|---|
1 | 1,048,576 | 10.32M | 21.90M | 2.12 | 1 | 1 |
8 | 131,072 | 58.55M | 93.73M | 1.60 | 5.67 | 4.28 |
Training performance: NVIDIA DGX-1 (8x V100 32GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX-1 with 8x V100 32GB GPUs.
The following table shows the best training throughput:
GPUs | Batch Size / GPU | Throughput - FP32 (samples / s) | Throughput - Mixed precision (samples /s) | Throughput Speedup (FP32 to Mixed precision) | Strong Scaling - FP32 | Strong scaling - Mixed precision |
---|---|---|---|---|---|---|
1 | 1,048,576 | 10.15M | 20.66M | 2.04 | 1 | 1 |
8 | 131,072 | 56.77M | 88.76M | 1.56 | 5.60 | 4.29 |
Training performance: NVIDIA DGX-2 (16x V100 32GB)
Our results were obtained by following the steps in the Quick Start Guide in the PyTorch 20.06-py3 NGC container on NVIDIA DGX-2 with 16x V100 32GB GPUs.
The following table shows the best training throughput:
GPUs | Batch Size / GPU | Throughput - FP32 (samples / s) | Throughput - Mixed precision (samples /s) | Throughput Speedup (FP32 to Mixed precision) | Strong Scaling - FP32 | Strong scaling - Mixed precision |
---|---|---|---|---|---|---|
1 | 1,048,576 | 10.83M | 22.18M | 2.05 | 1 | 1 |
8 | 131,072 | 60.33M | 95.58M | 1.58 | 5.57 | 4.31 |
16 | 65,536 | 83.00M | 120.1M | 1.45 | 7.67 | 5.41 |
Inference performance results
Inference performance: NVIDIA DGX A100 (1x A100 40GB)
Our results were obtained by running the inference.py
script in the PyTorch 20.06 NGC container on NVIDIA DGX A100 with 1x A100 GPU.
TF32
Batch size | Throughput Avg | Latency Avg | Latency 90% | Latency 95% | Latency 99% |
---|---|---|---|---|---|
1024 | 2.18e+06 | 0.00047 | 0.00045 | 0.00045 | 0.00045 |
4096 | 8.64e+06 | 0.00047 | 0.00046 | 0.00046 | 0.00046 |
16384 | 3.417e+07 | 0.00048 | 0.00046 | 0.00046 | 0.00046 |
65536 | 5.148e+07 | 0.00127 | 0.00125 | 0.00125 | 0.00125 |
262144 | 5.863e+07 | 0.00447 | 0.00444 | 0.00444 | 0.00444 |
1048576 | 6.032e+07 | 0.01738 | 0.01736 | 0.01736 | 0.01736 |
FP16
Batch size | Throughput Avg | Latency Avg | Latency 90% | Latency 95% | Latency 99% |
---|---|---|---|---|---|
1024 | 2.26e+06 | 0.00045 | 0.00044 | 0.00044 | 0.00044 |
4096 | 8.91e+06 | 0.00046 | 0.00044 | 0.00044 | 0.00044 |
16384 | 3.54e+07 | 0.00046 | 0.00045 | 0.00045 | 0.00045 |
65536 | 7.467e+07 | 0.00088 | 0.00086 | 0.00086 | 0.00086 |
262144 | 8.679e+07 | 0.00302 | 0.003 | 0.003 | 0.003 |
1048576 | 9.067e+07 | 0.01157 | 0.01154 | 0.01154 | 0.01154 |
Inference performance: NVIDIA DGX-1 (8x V100 16GB)
Our results were obtained by running the inference.py
script in the PyTorch 20.06 NGC container on NVIDIA DGX-1 with 1x V100 16GB GPU.
FP32
Batch size | Throughput Avg | Latency Avg | Latency 90% | Latency 95% | Latency 99% |
---|---|---|---|---|---|
1024 | 1.53e+06 | 0.00067 | 0.00065 | 0.00065 | 0.00065 |
4096 | 6.02e+06 | 0.00068 | 0.00067 | 0.00067 | 0.00067 |
16384 | 2.062e+07 | 0.00079 | 0.00076 | 0.00076 | 0.00076 |
65536 | 2.802e+07 | 0.00234 | 0.0023 | 0.0023 | 0.0023 |
262144 | 3.023e+07 | 0.00867 | 0.00865 | 0.00865 | 0.00865 |
1048576 | 3.015e+07 | 0.03478 | 0.03452 | 0.03452 | 0.03452 |
FP16
Batch size | Throughput Avg | Latency Avg | Latency 90% | Latency 95% | Latency 99% |
---|---|---|---|---|---|
1024 | 1.61e+06 | 0.00064 | 0.00062 | 0.00062 | 0.00062 |
4096 | 6.37e+06 | 0.00064 | 0.00063 | 0.00063 | 0.00063 |
16384 | 2.543e+07 | 0.00064 | 0.00063 | 0.00063 | 0.00063 |
65536 | 5.23e+07 | 0.00125 | 0.00121 | 0.00121 | 0.00121 |
262144 | 5.967e+07 | 0.00439 | 0.00437 | 0.00437 | 0.00437 |
1048576 | 5.998e+07 | 0.01748 | 0.01739 | 0.01739 | 0.01739 |
Release notes
Changelog
- January 22, 2018
- Initial release
- May, 2019
- Lower memory consumption (down from about 18GB to 10GB for batch size 1M on a single NVIDIA Tesla V100). Achieved by using an approximate method for generating negatives for training.
- Automatic Mixed Precision (AMP) with dynamic loss scaling instead of a custom mixed-precision optimizer.
- Performance numbers for NVIDIA DGX-2.
- Data loading code cleanup.
- Default container updated to PyTorch 19.05-py3.
- Updated README.md.
- June, 2019
- Updated performance tables.
- Default container changed to PyTorch 19.06-py3.
- Caching validation negatives between runs
- September, 2019
- Adjusting for API changes in PyTorch and APEX
- Checkpoints loading fix
- January, 2020
- DLLogger support added
- June, 2020
- Updated performance tables to include A100 results
Known issues
Scaling beyond 8 GPUs
Neural Collaborative Filtering is a relatively lightweight model that trains quickly with this relatively smaller dataset, ML-20m. Because of that, the high ratio of communication to computation makes it difficult to efficiently use more than 8 GPUs. Typically, this is not an issue because when using 8 GPUs with FP16 precision, the training is sufficiently fast. However, if you’d like to scale the training to 16 GPUs and beyond, you might try modifying the model so that the communication-computation ratio facilitates better scaling. This could be done, for example, by finding hyperparameters that enable using a larger batch size or by reducing the number of trainable parameters.
Memory usage
In the default settings, the additional memory beyond 16GB may not be fully utilized. This is because we set the default batch size for ML-20m dataset to 1M, which is too small to completely fill-up multiple 32GB GPUs. 1M is the batch size for which we experienced the best convergence on the ML-20m dataset. However, on other datasets, even faster performance can be possible by finding hyperparameters that work well for larger batches and leverage additional GPU memory.