Team NVIDIA has triumphed at the Amazon KDD Cup 2024, securing first place Friday across all five competition tracks.
The team - consisting of NVIDIANs Ahmet Erdem, Benedikt Schifferer, Chris Deotte, Gilberto Titericz, Ivan Sorokin and Simon Jegou - demonstrated its prowess in generative AI, winning in categories that included text generation, multiple-choice questions, name entity recognition, ranking, and retrieval.
The competition, themed Multi-Task Online Shopping Challenge for LLMs, asked participants to solve various challenges using limited datasets.
The new trend in LLM competitions is that they don't give you training data, said Deotte, a senior data scientist at NVIDIA. They give you 96 example questions - not enough to train a model - so we came up with 500,000 questions on our own.
Deotte explained that the NVIDIA team generated a variety of questions by writing some themselves, using a large language model to create others, and transforming existing e-commerce datasets.
Once we had our questions, it was straightforward to use existing frameworks to fine-tune a language model, he said.
The competition organizers hid the test questions to ensure participants couldn't exploit previously known answers. This approach encourages models that generalize well to any question about e-commerce, proving the model's capability to handle real-world scenarios effectively.
Despite these constraints, Team NVIDIA's innovative approach outperformed all competitors by using Qwen2-72B, a just-released LLM with 72 billion parameters, fine-tuned on eight NVIDIA A100 Tensor Core GPUs, and employing QLoRA, a technique for fine-tuning models with datasets.
About the KDD Cup 2024 The KDD Cup, organized by the Association for Computing Machinery's Special Interest Group on Knowledge Discovery and Data Mining, or ACM SIGKDD, is a prestigious annual competition that promotes research and development in the field.
This year's challenge, hosted by Amazon, focused on mimicking the complexities of online shopping with the goal of making it a more intuitive and satisfying experience using large language models. Organizers utilized the test dataset ShopBench - a benchmark that replicates the massive challenge for online shopping with 57 tasks and about 20,000 questions derived from real-world Amazon shopping data - to evaluate participants' models.
The ShopBench benchmark focused on four key shopping skills, along with a fifth all-in-one challenge:
Shopping Concept Understanding: Decoding complex shopping concepts and terminologies.
Shopping Knowledge Reasoning: Making informed decisions with shopping knowledge.
User Behavior Alignment: Understanding dynamic customer behavior.
Multilingual Abilities: Shopping across languages.
All-Around: Solving all tasks from the previous tracks in a unified solution.
NVIDIA's Winning Solution NVIDIA's winning solution involved creating a single model for each track.
The team fine-tuned the just-released Qwen2-72B model using eight NVIDIA A100 Tensor Core GPUs for approximately 24 hours. The GPUs provided fast and efficient processing, significantly reducing the time required for fine-tuning.
First, the team generated training datasets based on the provided examples and synthesized additional data using Llama 3 70B hosted on build.nvidia.com.
Next, they employed QLoRA (Quantized Low-Rank Adaptation), a training process using the data created in step one. QLoRA modifies a smaller subset of the model's weights, allowing efficient training and fine-tuning.
The model was then quantized - making it smaller and able to run on a system with a smaller hard drive and less memory - with AWQ 4-bit and used the vLLM inference library to predict the test datasets on four NVIDIA T4 Tensor Core GPUs within the time constraints.
This approach secured the top spot in each individual track and the overall first place in the competition-a clean sweep for NVIDIA for the second year in a row.
The team plans to submit a detailed paper on its solution next month and plans to present its findings at KDD 2024 in Barcelona.
More from Nvidia
17/09/2024
A new AI innovation hub for developers across Tunisia launched today in Novation City, a technology park that's designed to cultivate a vibrant, innovation ...
17/09/2024
At TwitchCon - a global convention for the Twitch livestreaming platform-livestreamers and content creators this week can experience the latest technologies for...
12/09/2024
Rise and shine - Capcom's latest action-adventure game, Dead Rising Deluxe Remaster, heads to the cloud at launch next week.
It's part of nine new titl...
11/09/2024
AI is transforming the broadcast industry by enhancing the way content is created, distributed and consumed - but integrating the technology can be challenging....
11/09/2024
Enterprises are looking for increasingly powerful compute to support their AI workloads and accelerate data processing. The efficiency gained can translate to b...
11/09/2024
To accelerate autonomous vehicle development and deployment timelines, Nuro announced today it will license its Nuro Driver autonomous driving system directly t...
09/09/2024
Companies in broadcast, sports and streaming are transitioning to software-defined infrastructure to benefit from flexible deployment and to more easily adopt t...
06/09/2024
Customer service departments across industries are facing increased call volumes, high customer service agent turnover, talent shortages and shifting customer e...
05/09/2024
Fall will be here soon, so leaf it to GeForce NOW to bring the games, with 19 joining the cloud in September.
Get started with the seven games available to str...
05/09/2024
The business transformations that generative AI brings come with risks that AI itself can help secure in a kind of flywheel of progress.
Companies who were qui...
04/09/2024
Volvo Cars' new, fully electric EX90 is making its way from the automaker's assembly line in Charleston, South Carolina, to dealerships around the U.S.
...
04/09/2024
Editor's note: This post is part of the AI Decoded series, which demystifies AI by making the technology more accessible, and showcases new hardware, softwa...
04/09/2024
Edward Mehr works where AI meets the anvil. The company he cofounded, Machina L...
04/09/2024
It all started at Berlin's Merantix venture studio in 2022, when Silviu Homo...
29/08/2024
Well before OpenAI upended the technology industry with its release of ChatGPT i...
29/08/2024
It's time to mana-fest the spirit of adventure with Square Enix's highly anticipated action role-playing game, Visions of Mana, launching today in the c...
28/08/2024
As enterprises race to adopt generative AI and bring new services to market, the demands on data center infrastructure have never been greater. Training large l...
28/08/2024
Editor's note: This post is part of the AI Decoded series, which demystifies AI by making the technology more accessible, and showcases new hardware, softwa...
27/08/2024
The initial wave of generative AI was driven by its use in internet services tha...
27/08/2024
Aiming at making the process faster and smarter, NVIDIA on Wednesday released th...
26/08/2024
Nations around the world are pursuing sovereign AI to produce artificial intelligence using their own computing infrastructure, data, workforce and business net...
23/08/2024
A deep technology conference for processor and system architects from industry a...
22/08/2024
Straight out of Gamescom, NVIDIA introduced GeForce NOW support for Xbox automat...
21/08/2024
Snowflake is using AI to help enterprises transform data into insights and applications. In this episode of NVIDIA's AI Podcast, host Noah Kravitz and Baris...
21/08/2024
Developers of generative AI typically face a tradeoff between model size and acc...
21/08/2024
Editor's note: This post is part of the AI Decoded series, which demystifies...
20/08/2024
At Gamescom, the world's biggest gaming expo, NVIDIA has once again pushed t...
20/08/2024
India is home to the globe's second-largest road network, spanning nearly 4 ...
20/08/2024
Picture this: NVIDIA and MediaTek are working together to make the industry's best gaming display technologies more accessible to gamers globally.
The comp...
20/08/2024
NVIDIA's first digital human technology small language model is being demons...
20/08/2024
Each week, GeForce NOW elevates cloud gaming by bringing top PC games and new up...
19/08/2024
As hurricanes, tornadoes and other extreme weather events occur with increased f...
15/08/2024
Time to be wowed: GeForce NOW members can now stream World of Warcraft on suppor...
14/08/2024
Editor's note: This post is part of the AI Decoded series, which demystifies...
13/08/2024
Bringing together the world's brightest minds and the latest accelerated computing technology leads to powerful breakthroughs that help tackle some of the b...
09/08/2024
The State of California today announced a first-of-its-kind AI education initiative with NVIDIA.
The public-private collaboration supports the state's goal...
08/08/2024
Editor's note: This blog was updated on Aug. 9 to reflect changes to the availability of Warhammer 40,000: Speed Freeks.'
This GFN Thursday marks 2,00...
08/08/2024
Silicon Valley's Figure has taken the wraps off of its next-generation Figur...
07/08/2024
Editor's note: This post is part of the AI Decoded series, which demystifies AI by making the technology more accessible, and showcases new hardware, softwa...
07/08/2024
Techbio is a field combining data, technology and biology to enhance scientific ...
06/08/2024
High school student Selin Alara Ornek is looking ahead - using machine learning and the NVIDIA Jetson platform for edge AI and robotics to create robot guide do...
06/08/2024
Editor's note: This post is part of our In the NVIDIA Studio series, which c...
01/08/2024
Members can choose their own adventure with GeForce NOW bringing 18 new games to the cloud in August - including Square Enix's fantasy role-playing game Vis...
31/07/2024
Enterprises are rapidly adopting generative AI, large language models (LLMs), ad...
31/07/2024
NVIDIA researchers used NVIDIA Edify, a multimodal architecture for visual generative AI, to build a detailed 3D desert landscape within a few minutes in a live...
31/07/2024
In celebration of Zoox's 10th anniversary, NVIDIA founder and CEO Jensen Huang recently joined the robotaxi company's CEO, Aicha Evans, and its cofounde...
31/07/2024
Editor's note: This post is part of the AI Decoded series, which demystifies...
29/07/2024
Designers and artists have new and improved ways to boost their productivity wit...
29/07/2024
Millions of people already use generative AI to assist in writing and learning. ...
Team NVIDIA has triumphed at the Amazon KDD Cup 2024, securing first place Friday across all five competition tracks.
