Stockfish NNUE

Home * Engines * Stockfish * NNUE



Stockfish NNUE, a Stockfish branch by Hisayori Noda aka Nodchip, which uses Efficiently Updatable Neural Networks - stylized as &#398;U&#1048;&#1048; or reversed as NNUE - to replace its standard evaluation. NNUE, introduced in 2018 by Yu Nasu , were previously successfully applied in Shogi evaluation functions embedded in a Stockfish based search, such as YaneuraOu , and Kristallweizen. YaneuraOu's author Motohiro Isozaki made an unbelievable prediction that NNUE can help to increase Stockfish strength by around 100 points, almost one year before revealing. In 2019, Nodchip incorporated NNUE into Stockfish 10 - as a proof of concept, and with the intention to give something back to the Stockfish community. After support and announcements by Henk Drost in May 2020 and subsequent enhancements, Stockfish NNUE was established and recognized. In summer 2020, with more people involved in testing and training, the computer chess community bursts out enthusiastically due to its rapidly raising playing strength with different networks trained using a mixture of supervised and reinforcement learning methods. Despite the approximately halved search speed, Stockfish NNUE became stronger than its original.

In August 2020, Fishtest revealed Stockfish NNUE was stronger than the classical one at least 80 Elo. In July 2020, the playing code of NNUE was put into the official Stockfish repository as a branch for further development and examination. In August that playing code merged to the master branch and become an official part of the engine. However, the training code still remained in Nodchip's repository. On September 02, 2020, Stockfish 12 was released with a huge jump in playing strength due to the introduction of NNUE and further tuning.

=NNUE Structure= The neural network consists of four layers. The input layer is heavily overparametrized, feeding in the board representation for all king placements per side. The efficiency of NNUE is due to incremental update of the input layer outputs in make and unmake move, where only a tiny fraction of its neurons need to be considered. The remaining three layers with 2x256x32, 32x32 and 32x1 weights are computational less expensive, hidden layer 1 and 2 with ReLu activation, best calculated using appropriate SIMD instructions performing fast 8-bit/16-bit integer vector arithmetic, like MMX, SSE2 or AVX2 on x86/x86-64, or, if available, AVX-512.

NNUE layers in action

Nodchip explained on Discord on June 2020 :

41,024 = 64 * 641. 64 comes from the number of the cells where king may exist. 641 = 64 * 5 * 2 + 1. 64 here comes from the number of the cells where a piece other than king may exist. 5 is the number of piece types other than king. 2 is the number of the colors, white and black. 1 is a captured piece. "+ 1" is BONA_PIECE_ZERO. Here "bona" means "bonanza" which is a popular computer shogi engine. It introduced the feature "p" for the first time. BonaPieces are contained in the evalList. It is updated by Position::do_move and Position::undo_move, and used by NNUE to calculate the network parameters between the input layer and the first hidden layer. About the calculation, the following text will be helpful. This text is sent to RocketMiningPoo on Twitter. "We add the i-th COLUMN of the W{0} to the z{0} for each i, where the i-th element is set to 1. And we subtract the i-th COULMN of the W{0} from the z{0} for each i, where the i-th element is set to 0. This operation is "accumulate" in your question." may be right... I hope that someone will double check.

=Network= All networks are built by some volunteers but not by any big community (differs from Leela Chess Zero) and can be uploaded into Fishtest for testing. Networks with good test results will be released officially on Fishtest website.

After long discussing the best way to publish networks with Stockfish, the developing team decided to embed the default network into Stockfish binaries, making sure NNUE always works as well as bringing more convenience to users.

=Hybrid= In August 2020 a new patch changed Stockfish NNUE into a hybrid engine: it uses NNUE evaluation only on quite balanced material positions, otherwise uses the classical one. It could speed up to 10% and gain 20 Elo. At that point, NNUE helped to increase already around 100 Elo for Stockfish. In the same month, Stockfish changed the default mode of using evaluation functions from classic to hybrid one, the last step to completely accept NNUE.

=Strong Points=

For Users

 * Runs with CPU only, and doesn't require expensive video cards, as well the need for installing drivers and 3rd specific libraries. Thus it is much easier to install (compared to engines using deep convolutional neural networks, such as Leela Chess Zero) and suitable for almost all modern computers. Using a GPU is even not practical for that small net - host-device-latency aka. kernel-launch-overhead to a then underemployed GPU are not sufficient for the intended NPS range
 * Releases with only one network (via UCI options), that help to delete users' confusion from finding, selecting and setting up. The network is selected carefully from Fishtest

For Developers

 * Requires small training sets. Some high score networks can be built with the effort of one or a few people within a few days. It doesn't require the massive computing from a supercomputer and/or from community
 * Doesn’t require complicated systems such as a sophisticated client-server model to train networks. Just a single binary from Nodchip’ repo is enough to train
 * The NNUE code is independent and can be separated easily from the rest and integrated to other engines

Being attracted by new advantages as well as being encouraged by some impressive successes, many developers joined or continued to work. The Official Stockfish repository shows the numbers of commits, ideas increased significantly after merging NNUE.

=Rotation vs Flip= Since the 9x9 Shogi board has a centered king file and castling is not known in Shogi, color flip versus180 degree rotate differs in a horizontal mirrored position from the other side's point of view, with otherwise identical playing options. The NNUE is trained and probed from the side to move point of view, where the used 180 degree rotation (xor 63 instead of 56) to flip sides looks rather strange for chess. i.e color flipping the black king from e8 to d1 rather than e1. Does it consider castling short to the queen side?

It is a little bit unclear, how that rotation rather than flip affects the playing strength and whether NNUE for chess suffers from associative visual agnosia. Maybe Fishtest needs to play many games with color-flipped openings, i.e. 1.e4 e5 and 1.e3 e5 2.e4, to look whether results differ or not. Anyway, a fix from rotate to flip has to be done from producer and consumer sides, and is likely to void some training sessions.

=Suggestions= In reply to Unai Corzo, Motohiro Isozaki aka Yaneurao, suggested 3 techniques that applied successfully to Shogi and can be brought back to Stockfish NNUE and may improve it another 100 - 200 Elo :
 * Optimizing all parameters together by stochastic optimization
 * Switching between multi-evaluation functions, according to game phases
 * Automatic generation of opening book on the fly

=See also=
 * Category: Neural Network Engines
 * Neural Networks
 * NNUE
 * Winter

=Forum Posts=

January ...

 * The Stockfish of shogi by Larry Kaufman, CCC, January 07, 2020 » Shogi
 * Stockfish NNUE by Henk Drost, CCC, May 31, 2020 » Stockfish
 * Stockfish NN release (NNUE) by Henk Drost, CCC, May 31, 2020
 * nnue-gui 1.0 released by Norman Schmidt, CCC, June 17, 2020
 * stockfish-NNUE as grist for SF development? by Warren D. Smith, FishCooking, June 21, 2020

July

 * How strong is Stockfish NNUE compared to Leela.. by OmenhoteppIV, LCZero Forum, July 13, 2020 » Leela Chess Zero
 * Can the sardine! NNUE clobbers SF by Henk Drost, CCC, July 16, 2020
 * End of an era? by Michel Van den Bergh, FishCooking, July 20, 2020
 * Sergio Vieri second net is out by Sylwy, CCC, July 21, 2020
 * NNUE accessible explanation by Martin Fierz, CCC, July 21, 2020
 * Re: NNUE accessible explanation by Jonathan Rosenthal, CCC, July 23, 2020
 * Re: NNUE accessible explanation by Jonathan Rosenthal, CCC, July 24, 2020
 * Re: NNUE accessible explanation by Jonathan Rosenthal, CCC, August 03, 2020


 * Stockfisch NNUE macOS binary requested by Steppenwolf, CCC, July 25, 2020
 * Stockfish NNUE by Lion, CCC, July 25, 2020
 * 7000 games testrun of SFnnue sv200724_0123 finished by Stefan Pohl, FishCooking, July 26, 2020
 * SF-NNUE going forward... by Zenmastur, CCC, July 27, 2020
 * New sf+nnue play-only compiles by Norman Schmidt, CCC, July 27, 2020
 * Stockfish+NNUEsv +80 Elo vs Stockfish 17jul !! by Kris, Rybka Forum, July 28, 2020
 * LC0 vs. NNUE - some tech details... by Srdja Matovic, CCC, July 29, 2020 » Lc0
 * Stockfish NNUE and testsuites by Jouni Uski, CCC, July 29, 2020
 * Stockfish NNue [download ] by Ed Schröder, CCC, July 30, 2020

August

 * Repository for Stockfish+NNUE Android Builds by AdminX, CCC, August 02, 2020
 * SF NNUE Problem by Stephen Ham, CCC, August 03, 2020
 * Re: NNUE accessible explanation by Jonathan Rosenthal, CCC, August 03, 2020 » NNUE accessible explanation
 * [NNUE Worker update on fishtest ] by Joost VandeVondele, FishCooking, August 03, 2020
 * this will be the merge of a lifetime : SF 80 Elo+ by MikeB, CCC, August 04, 2020
 * Re: this will be the merge of a lifetime : SF 80 Elo+ by Henk Drost, CCC, August 04, 2020


 * You can now look inside NNUE and look at its Per square value estimation by Henk Drost, CCC, August 04, 2020
 * Is this SF NN almost like 20 MB book? by Jouni Uski, CCC, August 04, 2020
 * NNUE evaluation merged in master by Joost VandeVondele, FishCooking, August 06, 2020
 * Stockfish NNUE style by Rowen, CCC, August 08, 2020
 * SF NNUE training questions by Jouni Uski, CCC, August 10, 2020
 * Neural Networks weights type by Fabio Gobbato, CCC, August 13, 2020
 * Don't understand NNUE by Lucasart, CCC, August 14, 2020
 * SF+NNUE reach the ceiling? by Corres, CCC, August 27, 2020
 * The most stupid idea by the Stockfish Team by Damir, CCC, August 30, 2020

September

 * Stockfish 12 by Joost VandeVondele, FishCooking, September 02, 2020
 * Stockfish 12 is released today! by Nay Lin Tun, CCC, September 02, 2020
 * Stockfish 12 has arrived! by daniel71, CCC, September 02, 2020
 * AVX2 optimized SF+NNUE and processor temperature by corres, CCC, September 05, 2020 » AVX2

October ...

 * BONA_PIECE_ZERO by elcabesa, CCC, October 04, 2020
 * SF NNUE/Classical by Fauzi, FishCooking, October 05, 2020

=External Links=

Basics

 * Stockfish NNUE – The Complete Guide, June 19, 2020 (Japanese and English)
 * 3 technologies in shogi AI that could be used for chess AI by Motohiro Isozaki, August 21, 2020
 * Stockfish NNUE Wiki
 * NNUE merge · Issue #2823 · official-stockfish/Stockfish · GitHub by Joost VandeVondele, July 25, 2020
 * Stockfish Evaluation Guide

Source

 * GitHub - Official-stockfish
 * GitHub - nodchip/Stockfish: UCI chess engine by Nodchip
 * GitHub - vondele/Stockfish at nnue-player-wip by Joost VandeVondele
 * GitHub - tttak/Stockfish: UCI chess engine
 * GitHub - joergoster/Stockfish-NNUE: UCI Chess engine Stockfish with an Efficiently Updatable Neural-Network-based evaluation function hosted by Jörg Oster
 * GitHub - FireFather/sf-nnue: Stockfish NNUE (efficiently updateable neural network) by Norman Schmidt
 * GitHub - FireFather/nnue-gui: basic windows application for using nodchip's stockfish-nnue software by Norman Schmidt

Networks

 * Neural Net download and statistics
 * Index of /~sergio-v/nnue by Sergio Vieri

Rating

 * Regression Tests
 * Stockfish 12 64-bit in CCRL Blitz
 * Stockfish+NNUE 150720 64-bit 4CPU in CCRL Blitz

Misc

 * Stockfish from Wikipedia
 * Nue from Wikipedia
 * Hiromi - Spectrum, 2019, YouTube Video

=References= Up one Level