• Stars
    star
    254
  • Rank 160,264 (Top 4 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created over 3 years ago
  • Updated 3 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Falken provides developers with a service that allows them to train AI that can play their games

Falken

Main branch Latest release

NOTE: This is not an officially supported Google product.

Falken provides developers with a service that allows them to train AI that can play their games. Unlike traditional RL frameworks that learn through rewards or batches of offline training, Falken is based on training AI via realtime, human interactions.

Because human time is precious, the system is designed to seamlessly transition between humans playing the game (training) and the AI playing the game (inference). This allows us to support a variety of training styles, from providing a large batch of demonstrations up front to providing “just in time” demonstrations when the AI encounters a state it does not know how to handle.

Behind the scenes, Falken automatically handles logging of experience, separates out human demonstrations, trains new models, serves the best models for on-device inference, and maintains a version history.

Components

This project consists of:

  • A local service that collects experience, trains and serves models.
  • A C++ SDK that can be integrated into C++ games to use the service and an example environment, Hello Falken that demonstrates use of the C++ SDK.
  • A Unity SDK that can be integrated into Unity games to use the service and example environments with Hello Falken being the most simple to get started.
  • A simple web dashboard that connects to the service to provides a way to visualize traces from sessions for debugging purposes.

Getting started

The fastest way to ensure that everything is working properly is to launch a local service then build and run Hello Falken environment.

In C++:

In Unity:

Concepts

Stand up and Initialize Falken

The first step in using Falken is to launch the service and connect to the service from the client SDK (e.g C++ or Unity) using your JSON configuration. All subsequent API calls from the client will automatically use this information.

Define Actions

Actions allow you to define how Falken’s AI interacts with your game and how it learns from humans who play it. In Falken, Actions can be either continuous or discrete.

  • Continuous Actions are used to represent analog controls like a joystick that controls aiming or an analog trigger that controls the gas pedal in a car. All continuous actions are required to include a developer defined min and max value.
  • Discrete Actions can be used to represent simple buttons (like whether the fire button is pressed) or discrete selections (like which weapon to equip). We generally recommend developers create Actions which represent logical interactions (jump, shoot, move, etc) rather than physical inputs (A button, left stick). As such, you may have more Actions in Falken than there are physical inputs for your game.

Define Observations

Observations allow you to define how Falken’s AI perceives your game both while it is playing and while it is “watching” a human play. Falken’s Observations are based on the Entity/Attribute model, where an Entity is simply a container for attributes, each of which have a corresponding name, type, and value.

  • Position: All Entities include an optional, pre-defined 3D position attribute (vector). If used, this position should generally be in world space.
  • Rotation: All Entities include an optional, pre-defined 3D rotation attribute (quaternion). If used, this rotation should generally be in world space.
  • Number: Number attributes represent quantities in the world that can be counted, either as whole numbers (ints) or real numbers (floats). All Number attributes are required to have a developer-defined minimum and maximum value.
  • Category: Categorical attributes represent values with discrete states, like a simple bool or the entries in an enum.
  • Feelers: Feelers represent a set of evenly spaced samples reaching out from an entity. They are commonly used to sense the 3D environment around an entity (usually the player).

Defining the best Observations for your game can be tricky and can impact Falken's performance, so we recommend starting with the bare minimum a human would require to play a game and slowly refining the set of Observations to improve efficacy. In practice we've found that simple observations can yield much better performance.

Create a Brain

In Falken, a Brain represents the idea of the thing that observes, learns, and acts. It’s also helpful to think about each Brain as being designed to accomplish a specific kind of task, like completing a lap in a racing game or defeating enemies while moving through a level in an FPS.

When creating a Brain, you provide a specific set of Actions and a specific set of Observations that define how the brain perceives the world and what actions it can take.

Start a Session. Start an Episode.

In Falken, a Session represents a period of time during which Falken was learning how to play your game, playing your game, or both. Sessions are composed of one or more Episodes, which represent a sequence of interactions with a clear beginning, middle, and end (like starting a race, driving around the track, and crossing the finish line).

Falken supports a number of different Session types, including:

  • Interactive Training: In this mode, Falken learns by watching a human play the game. Players can teach Falken by playing entire Episodes from beginning to end or they can let Falken play and only take over to suggest a correction in behavior whenever Falken makes a mistake.
  • Evaluation: In this mode, Falken plays the game autonomously in order to determine the best model that can be produced for your game given the data provided in earlier Interactive Training mode.
  • Inference: In this mode, Falken simply plays your game as much as you like and intentionally disables all learning features. This guarantees that you get consistent results, and is ideal for testing your game at scale.

All Sessions also require you to define a maximum number of Steps per Episode. You’ll want to set this number to be a comfortable amount more than one a human would require. For example, if you call Episode::Step() every frame, your game runs at 30fps, and a human would take 300 seconds to complete an episode, then you should set a max steps to about 9000 or 12000 to ensure the AI has enough time to complete the task. If the number of steps is too high then Falken will take longer to detect when an agent is stuck when using an evaluation session.

Step the Episode

Every time you want Falken to learn from a human or interact with your game, you need to need to call Step(). Step() always operates on a set of Observations from your game. If a human is playing, then you also provide the Actions that the human performed, and Falken will learn from them. If you want Falken to play your game, don’t pass in any Actions to Step() but instead use the Actions returned by Step() and apply them to your game as if a human had performed them.

For most games, you’ll call Falken’s Step() every time you update your game simulation. However, some games might want to call Step() less often, like when the player takes a turn or performs an important action.

End the Episode. Reset the Game.

Episodes in Falken end when the player succeeds, fails, or the maximum number of steps has been exceeded. You provide this context when you end the Episode.

  • Success: The player (human or AI) has successfully accomplished the goal (e.g. completed a lap or finished the level).
  • Failure: The player (human or AI) failed to accomplish the goal (e.g. car crashed or player was killed by an enemy).
  • Gave Up: The episode reached its max steps without succeeding or failing.
  • Aborted: The episode was unexpectedly terminated in the middle.

Most Sessions involve many Episodes (either for training Falken or testing your game), so when an Episode completes, you’ll need to reset your game state before you bring the next Episode. Falken leaves the definition of this reset up to you and does not require that every Episode in a Session start with identical state. In general, you want each episode to start with similar but not identical state (similar to what many games do for human players) as this helps Falken most effectively learn from and test your game.

Ending one Session. Starting a new Session.

Once you are done with a Session, you simply Stop it. If an Episode is still active when a Session is stopped, the Episode will be Aborted.

When you Stop a learning-type session (InteractiveTraining or Evaluation), the Brain makes a record of what it has learned and "gets smarter". Additional Sessions created with this brain will start from this more intelligent state and use that as the basis for additional learning or playing.

More Repositories

1

bert

TensorFlow code and pre-trained models for BERT
Python
37,769
star
2

google-research

Google Research
Jupyter Notebook
33,759
star
3

tuning_playbook

A playbook for systematically maximizing the performance of deep learning models.
26,593
star
4

vision_transformer

Jupyter Notebook
10,251
star
5

text-to-text-transfer-transformer

Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
Python
6,099
star
6

arxiv-latex-cleaner

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
Python
5,233
star
7

simclr

SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners
Jupyter Notebook
3,937
star
8

multinerf

A Code Release for Mip-NeRF 360, Ref-NeRF, and RawNeRF
Python
3,612
star
9

timesfm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
Python
3,576
star
10

scenic

Scenic: A Jax Library for Computer Vision Research and Beyond
Python
3,295
star
11

football

Check out the new game server:
Python
3,260
star
12

albert

ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
Python
3,209
star
13

frame-interpolation

FILM: Frame Interpolation for Large Motion, In ECCV 2022.
Python
2,818
star
14

t5x

Python
2,656
star
15

electra

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators
Python
2,325
star
16

kubric

A data generation pipeline for creating semi-realistic synthetic multi-object videos with rich annotations such as instance segmentation masks, depth maps, and optical flow.
Jupyter Notebook
2,312
star
17

big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
Jupyter Notebook
2,219
star
18

uda

Unsupervised Data Augmentation (UDA)
Python
2,131
star
19

language

Shared repository for open-sourced projects from the Google AI Language team.
Python
1,605
star
20

pegasus

Python
1,600
star
21

dex-lang

Research language for array processing in the Haskell/ML family
Haskell
1,581
star
22

torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
Python
1,548
star
23

parti

1,538
star
24

big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
Python
1,504
star
25

FLAN

Python
1,460
star
26

robotics_transformer

Python
1,337
star
27

disentanglement_lib

disentanglement_lib is an open-source library for research on learning disentangled representations.
Python
1,311
star
28

multilingual-t5

Python
1,197
star
29

circuit_training

Python
1,151
star
30

tapas

End-to-end neural table-text understanding models.
Python
1,143
star
31

planet

Learning Latent Dynamics for Planning from Pixels
Python
1,134
star
32

mixmatch

Python
1,130
star
33

deduplicate-text-datasets

Rust
1,104
star
34

fixmatch

A simple method to perform semi-supervised learning with limited data.
Python
1,094
star
35

morph-net

Fast & Simple Resource-Constrained Learning of Deep Network Structure
Python
1,016
star
36

maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.
Python
996
star
37

deeplab2

DeepLab2 is a TensorFlow library for deep labeling, aiming to provide a unified and state-of-the-art TensorFlow codebase for dense pixel labeling tasks.
Python
995
star
38

batch-ppo

Efficient Batched Reinforcement Learning in TensorFlow
Python
963
star
39

augmix

AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty
Python
951
star
40

magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer
Python
947
star
41

pix2seq

Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)
Jupyter Notebook
865
star
42

seed_rl

SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's architecture.
Python
793
star
43

meta-dataset

A dataset of datasets for learning to learn from few examples
Jupyter Notebook
762
star
44

noisystudent

Code for Noisy Student Training. https://arxiv.org/abs/1911.04252
Python
751
star
45

rliable

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.
Jupyter Notebook
747
star
46

recsim

A Configurable Recommender Systems Simulation Platform
Python
739
star
47

jax3d

Python
733
star
48

long-range-arena

Long Range Arena for Benchmarking Efficient Transformers
Python
719
star
49

lottery-ticket-hypothesis

A reimplementation of "The Lottery Ticket Hypothesis" (Frankle and Carbin) on MNIST.
Python
706
star
50

federated

A collection of Google research projects related to Federated Learning and Federated Analytics.
Python
675
star
51

bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.
Python
651
star
52

prompt-tuning

Original Implementation of Prompt Tuning from Lester, et al, 2021
Python
642
star
53

nasbench

NASBench: A Neural Architecture Search Dataset and Benchmark
Python
641
star
54

neuralgcm

Hybrid ML + physics model of the Earth's atmosphere
Python
641
star
55

xtreme

XTREME is a benchmark for the evaluation of the cross-lingual generalization ability of pre-trained multilingual models that covers 40 typologically diverse languages and includes nine tasks.
Python
631
star
56

lasertagger

Python
606
star
57

sound-separation

Python
603
star
58

pix2struct

Python
587
star
59

vmoe

Jupyter Notebook
569
star
60

dreamer

Dream to Control: Learning Behaviors by Latent Imagination
Python
568
star
61

robopianist

[CoRL '23] Dexterous piano playing with deep reinforcement learning.
Python
562
star
62

omniglue

Code release for CVPR'24 submission 'OmniGlue'
Python
561
star
63

fast-soft-sort

Fast Differentiable Sorting and Ranking
Python
561
star
64

ravens

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.
Python
560
star
65

sam

Python
551
star
66

batch_rl

Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games
Python
521
star
67

bigbird

Transformers for Longer Sequences
Python
518
star
68

tensor2robot

Distributed machine learning infrastructure for large-scale robotics research
Python
483
star
69

byt5

Python
477
star
70

adapter-bert

Python
476
star
71

mint

Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.
Python
465
star
72

leaf-audio

LEAF is a learnable alternative to audio features such as mel-filterbanks, that can be initialized as an approximation of mel-filterbanks, and then be trained for the task at hand, while using a very small number of parameters.
Python
446
star
73

robustness_metrics

Jupyter Notebook
442
star
74

maxvit

[ECCV 2022] Official repository for "MaxViT: Multi-Axis Vision Transformer". SOTA foundation models for classification, detection, segmentation, image quality, and generative modeling...
Jupyter Notebook
436
star
75

receptive_field

Compute receptive fields of your favorite convnets
Python
434
star
76

maskgit

Official Jax Implementation of MaskGIT
Jupyter Notebook
429
star
77

weatherbench2

A benchmark for the next generation of data-driven global weather models.
Python
420
star
78

l2p

Learning to Prompt (L2P) for Continual Learning @ CVPR22 and DualPrompt: Complementary Prompting for Rehearsal-free Continual Learning @ ECCV22
Python
408
star
79

distilling-step-by-step

Python
407
star
80

ssl_detection

Semi-supervised learning for object detection
Python
398
star
81

nerf-from-image

Shape, Pose, and Appearance from a Single Image via Bootstrapped Radiance Field Inversion
Python
377
star
82

computation-thru-dynamics

Understanding computation in artificial and biological recurrent networks through the lens of dynamical systems.
Jupyter Notebook
369
star
83

tf-slim

Python
368
star
84

realworldrl_suite

Real-World RL Benchmark Suite
Python
341
star
85

python-graphs

A static analysis library for computing graph representations of Python programs suitable for use with graph neural networks.
Python
325
star
86

rigl

End-to-end training of sparse deep neural networks with little-to-no performance loss.
Python
314
star
87

task_adaptation

Python
310
star
88

self-organising-systems

Jupyter Notebook
308
star
89

ibc

Official implementation of Implicit Behavioral Cloning, as described in our CoRL 2021 paper, see more at https://implicitbc.github.io/
Python
306
star
90

tensorflow_constrained_optimization

Python
300
star
91

syn-rep-learn

Learning from synthetic data - code and models
Python
294
star
92

arco-era5

Recipes for reproducing Analysis-Ready & Cloud Optimized (ARCO) ERA5 datasets.
Python
291
star
93

vdm

Jupyter Notebook
291
star
94

rlds

Jupyter Notebook
284
star
95

exoplanet-ml

Machine learning models and utilities for exoplanet science.
Python
283
star
96

retvec

RETVec is an efficient, multilingual, and adversarially-robust text vectorizer.
Jupyter Notebook
281
star
97

sparf

This is the official code release for SPARF: Neural Radiance Fields from Sparse and Noisy Poses [CVPR 2023-Highlight]
Python
279
star
98

tensorflow-coder

Python
275
star
99

lm-extraction-benchmark

Python
270
star
100

language-table

Suite of human-collected datasets and a multi-task continuous control benchmark for open vocabulary visuolinguomotor learning.
Jupyter Notebook
260
star