Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/lucasem/program-induction

A library for program induction and learning representations.
https://github.com/lucasem/program-induction

bayesian-inference genetic-programming lambda-calculus pcfg program-induction representation-learning

Last synced: 2 months ago
JSON representation

A library for program induction and learning representations.

Awesome Lists containing this project

README 

        
# program-induction



[![crates.io](https://img.shields.io/crates/v/programinduction.svg)](https://crates.io/crates/programinduction)

[![docs.rs](https://docs.rs/programinduction/badge.svg)](https://docs.rs/programinduction)

[![CI](https://github.com/lorepozo/program-induction/actions/workflows/ci.yaml/badge.svg)](https://github.com/lorepozo/program-induction/actions/workflows/ci.yaml)



A library for program induction and learning representations.



Implements Bayesian program learning and genetic programming.

See the [docs](https://docs.rs/programinduction) for more information.



## Installation



Install [rust](https://rust-lang.org) and ensure you're up to date (`rustup update`).

In a new or existing project, add the following to your `Cargo.toml`:



```toml

[dependencies]

programinduction = "0.9"

# many examples also depend on polytype for the tp! and ptp! macros:

polytype = "7.0"

```



The documentation requires a custom HTML header to include KaTeX for math

support. This isn't supported by `cargo doc`, so to build the documentation

you may use:



```sh

cargo rustdoc -- --html-in-header rustdoc-include-katex-header.html

```



## Usage



Specify a probabilistic context-free grammar (PCFG; see `pcfg::Grammar`) and

induce a sentence that matches an example:



```rust

use polytype::tp;

use programinduction::pcfg::{task_by_evaluation, Grammar, Rule};

use programinduction::{ECParams, EC};



fn evaluate(name: &str, inps: &[i32]) -> Result {

    match name {

        "0" => Ok(0),

        "1" => Ok(1),

        "plus" => Ok(inps[0] + inps[1]),

        _ => unreachable!(),

    }

}



fn main() {

    let g = Grammar::new(

        tp!(EXPR),

        vec![

            Rule::new("0", tp!(EXPR), 1.0),

            Rule::new("1", tp!(EXPR), 1.0),

            Rule::new("plus", tp!(@arrow[tp!(EXPR), tp!(EXPR), tp!(EXPR)]), 1.0),

        ],

    );

    let ec_params = ECParams {

        frontier_limit: 1,

        search_limit_timeout: None,

        search_limit_description_length: Some(8.0),

    };

    // task: the number 4

    let task = task_by_evaluation(&evaluate, &4, tp!(EXPR));



    let frontiers = g.explore(&ec_params, &[task]);

    let sol = &frontiers[0].best_solution().unwrap().0;

    println!("{}", g.display(sol));

}

```



The Exploration-Compression (EC) algorithm iteratively learns a better

representation by finding common structure in induced programs. We can run

the EC algorithm with a polymorphically-typed lambda calculus representation

`lambda::Language` in a Boolean circuit domain:



```rust

use polytype::{ptp, tp};

use programinduction::{domains, lambda, ECParams, EC};



fn main() {

    // circuit DSL

    let dsl = lambda::Language::uniform(vec![

        // NAND takes two bools and returns a bool

        ("nand", ptp!(@arrow[tp!(bool), tp!(bool), tp!(bool)])),

    ]);

    // parameters

    let lambda_params = lambda::CompressionParams::default();

    let ec_params = ECParams {

        frontier_limit: 1,

        search_limit_timeout: Some(std::time::Duration::new(1, 0)),

        search_limit_description_length: None,

    };

    // randomly sample 250 circuit tasks

    let rng = &mut rand::thread_rng();

    let tasks = domains::circuits::make_tasks(rng, 250);



    // one iteration of EC:

    let (new_dsl, _solutions) = dsl.ec(&ec_params, &lambda_params, &tasks);

    // print the new concepts it invented, based on common structure:

    for (expr, _, _) in &new_dsl.invented {

        println!("invented {}", new_dsl.display(expr))

        // one of the inventions was "(λ (nand $0 $0))",

        // which is the common and useful NOT operation!

    }

}

```



You may have noted the above use of `domains::circuits`. Some domains are

already implemented for you. Currently, this only consists of _circuits_ and

_strings_.



## TODO



(you could be the one who does one of these!)



- [x] First-class function evaluation within Rust (and remove lisp

      interpreters).

- [x] Add task generation function in `domains::strings`

- [x] Fallible evaluation (e.g. see how `domains::strings` handles `slice`).

- [x] Lazy evaluation.

- [x] `impl GP for pcfg::Grammar` is not yet complete.

- [ ] Eta-long sidestepping (so `f` gets enumerated instead of `(λ (f $0))`)

- [ ] Consolidate lazy/non-lazy evaluation (for ergonomics).

- [ ] Permit non-`&'static str`-named `Type`/`TypeScheme`.

- [ ] Ability to include recursive primitives in `lambda` representation.

- [ ] Faster lambda calculus evaluation (less cloning; bubble up whether

      beta reduction happened rather than ultimate equality comparison).

- [ ] PCFG compression is currently only estimating parameters, not actually

      learning pieces of programs. An [adaptor

      grammar](http://cocosci.berkeley.edu/tom/papers/adaptornips.pdf)

      approach seems like a good direction to go, perhaps minus the Bayesian

      non-parametrics.

- [ ] Add more learning traits (like `EC` or `GP`)

- [ ] Add more representations

- [ ] Add more domains